diff --git a/docs/notebooks/sinusoidal/sinusoidal_example.ipynb b/docs/notebooks/sinusoidal/sinusoidal_example.ipynb index 4eabc8c..c73ae3b 100644 --- a/docs/notebooks/sinusoidal/sinusoidal_example.ipynb +++ b/docs/notebooks/sinusoidal/sinusoidal_example.ipynb @@ -9,7 +9,7 @@ }, { "cell_type": "code", - "execution_count": 1, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ @@ -20,17 +20,9 @@ }, { "cell_type": "code", - "execution_count": 2, + "execution_count": null, "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "\n" - ] - } - ], + "outputs": [], "source": [ "# Update the `LC_METHODS` registration dictionary\n", "update_lc_subclasses()\n", @@ -41,22 +33,9 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": null, "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - " FieldMJD LCA Period Time0\n", - "0 60277.351867 1 0.001 60277.351867\n", - "1 60289.319749 1 0.001 60277.351867\n", - "2 60289.330920 1 0.001 60277.351867\n", - "3 60292.334497 1 0.001 60277.351867\n", - "4 60292.346208 1 0.001 60277.351867\n" - ] - } - ], + "outputs": [], "source": [ "# make a small pandas dataframe with ~5 rows\n", "import pandas as pd\n", @@ -74,22 +53,9 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": null, "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "0 -0.340932\n", - "1 -0.934792\n", - "2 0.186596\n", - "3 -0.026723\n", - "4 0.737016\n", - "dtype: float64\n" - ] - } - ], + "outputs": [], "source": [ "# instantiate the sinusoidal class\n", "lc_model = LC_METHODS['sinusoidal']()\n", @@ -118,8 +84,7 @@ "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.10.12" - }, - "orig_nbformat": 4 + } }, "nbformat": 4, "nbformat_minor": 2 diff --git a/src/sorcha_community_utils/sinusoidal/sinusoidal_lightcurve.py b/src/sorcha_community_utils/sinusoidal/sinusoidal_lightcurve.py index edbe265..20bd52f 100644 --- a/src/sorcha_community_utils/sinusoidal/sinusoidal_lightcurve.py +++ b/src/sorcha_community_utils/sinusoidal/sinusoidal_lightcurve.py @@ -4,6 +4,7 @@ import pandas as pd import numpy as np + class SinusoidalLightCurve(AbstractLightCurve): """ Note: assuming sinusoidal in magnitude instead of flux. Maybe not call LCA? diff --git a/tests/sinusoidal/test_sinusoidal.py b/tests/sinusoidal/test_sinusoidal.py index 2183abc..9fef57b 100644 --- a/tests/sinusoidal/test_sinusoidal.py +++ b/tests/sinusoidal/test_sinusoidal.py @@ -1,17 +1,17 @@ from sorcha_community_utils.sinusoidal.sinusoidal_lightcurve import SinusoidalLightCurve -import numpy as np import pandas as pd + def test_sinusoidal_lightcurve_name(): - assert 'sinusoidal' == SinusoidalLightCurve.name_id() + assert "sinusoidal" == SinusoidalLightCurve.name_id() def test_compute_simple(): data_dict = { - 'FieldMJD': [0], - 'LCA': [1], - 'Period': [1], - 'Time0': [0], + "FieldMJD": [0], + "LCA": [1], + "Period": [1], + "Time0": [0], } df = pd.DataFrame.from_dict(data_dict)