Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Move dictionaries needed for testing to new file #466

Merged
merged 1 commit into from
Oct 8, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion stellarphot/gui_tools/tests/test_seeing_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
PhotometryWorkingDirSettings,
settings_files, # This import is needed for mocking
)
from stellarphot.settings.tests.test_models import (
from stellarphot.settings.constants import (
TEST_CAMERA_VALUES,
)

Expand Down
2 changes: 1 addition & 1 deletion stellarphot/photometry/tests/test_profiles.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from stellarphot.photometry import CenterAndProfile, find_center
from stellarphot.photometry.tests.fake_image import make_gaussian_sources_image
from stellarphot.settings import Camera
from stellarphot.settings.tests.test_models import TEST_CAMERA_VALUES
from stellarphot.settings.constants import TEST_CAMERA_VALUES

SHAPE = (300, 300)
RANDOM_SEED = 1230971
Expand Down
2 changes: 1 addition & 1 deletion stellarphot/settings/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
# For conveniences, provide the JSON schema and an example of PhotometrySettings.
photometry_settings_schema = PhotometrySettings.model_json_schema() # noqa: F405

from .tests.test_models import TEST_PHOTOMETRY_SETTINGS
from .constants import TEST_PHOTOMETRY_SETTINGS

photometry_settings_example = PhotometrySettings( # noqa: F405
**TEST_PHOTOMETRY_SETTINGS
Expand Down
95 changes: 95 additions & 0 deletions stellarphot/settings/constants.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,95 @@
# Though these constants are used in the tests they are needed in a few
# non-test places too. Putting them here ensures they can be imported
# without needing test dependencies.

TEST_APERTURE_SETTINGS = dict(radius=5, gap=10, annulus_width=15, fwhm=3.2)

TEST_CAMERA_VALUES = dict(
data_unit="adu",
gain="2.0 electron / adu",
name="test camera",
read_noise="10.0 electron",
dark_current="0.01 electron / s",
pixel_scale="0.563 arcsec / pix",
max_data_value="50000.0 adu",
)

TEST_EXOPLANET_SETTINGS = dict(
epoch={
"jd1": 0.0,
"jd2": 0.0,
"format": "jd",
"scale": "utc",
"precision": 3,
"in_subfmt": "*",
"out_subfmt": "*",
},
period="0.0 min",
identifier="a planet",
coordinate={
"ra": "0d00m00s",
"dec": "0d00m00s",
"representation_type": "spherical",
"frame": "icrs",
},
depth=0,
duration="0.0 min",
)

TEST_OBSERVATORY_SETTINGS = dict(
name="test observatory",
longitude="43d00m00s",
latitude="45d00m00s",
elevation="311.0 m",
AAVSO_code="test",
TESS_telescope_code="tess test",
)

# The first setting here is required, the rest are optional. The optional
# settings below are different than the defaults in the model definition.
TEST_PHOTOMETRY_OPTIONS = dict(
include_dig_noise=False,
reject_too_close=False,
reject_background_outliers=False,
fwhm_by_fit=False,
method="center",
)

TEST_PASSBAND_MAP = dict(
name="Example map",
your_filter_names_to_aavso=[
dict(
your_filter_name="V",
aavso_filter_name="V",
),
dict(
your_filter_name="B",
aavso_filter_name="B",
),
dict(
your_filter_name="rp",
aavso_filter_name="SR",
),
],
)

TEST_LOGGING_SETTINGS = dict(
logfile="test.log",
console_log=False,
)

TEST_SOURCE_LOCATION_SETTINGS = dict(
shift_tolerance=5,
source_list_file="test.ecsv",
use_coordinates="pixel",
)

TEST_PHOTOMETRY_SETTINGS = dict(
camera=TEST_CAMERA_VALUES,
observatory=TEST_OBSERVATORY_SETTINGS,
photometry_apertures=TEST_APERTURE_SETTINGS,
source_location_settings=TEST_SOURCE_LOCATION_SETTINGS,
photometry_optional_settings=TEST_PHOTOMETRY_OPTIONS,
passband_map=TEST_PASSBAND_MAP,
logging_settings=TEST_LOGGING_SETTINGS,
)
12 changes: 6 additions & 6 deletions stellarphot/settings/tests/test_custom_widgets.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,12 @@
settings_files,
ui_generator,
)
from stellarphot.settings.constants import (
TEST_CAMERA_VALUES,
TEST_OBSERVATORY_SETTINGS,
TEST_PASSBAND_MAP,
TEST_PHOTOMETRY_SETTINGS,
)
from stellarphot.settings.custom_widgets import (
ChooseOrMakeNew,
Confirm,
Expand All @@ -31,12 +37,6 @@
Spinner,
_add_saving_to_widget,
)
from stellarphot.settings.tests.test_models import (
TEST_CAMERA_VALUES,
TEST_OBSERVATORY_SETTINGS,
TEST_PASSBAND_MAP,
TEST_PHOTOMETRY_SETTINGS,
)


# See test_settings_file.TestSavedSettings for a detailed description of what the
Expand Down
103 changes: 11 additions & 92 deletions stellarphot/settings/tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,17 @@
from pydantic import ValidationError

from stellarphot.settings import ui_generator
from stellarphot.settings.constants import (
TEST_APERTURE_SETTINGS,
TEST_CAMERA_VALUES,
TEST_EXOPLANET_SETTINGS,
TEST_LOGGING_SETTINGS,
TEST_OBSERVATORY_SETTINGS,
TEST_PASSBAND_MAP,
TEST_PHOTOMETRY_OPTIONS,
TEST_PHOTOMETRY_SETTINGS,
TEST_SOURCE_LOCATION_SETTINGS,
)
from stellarphot.settings.models import (
Camera,
Exoplanet,
Expand All @@ -23,98 +34,6 @@
SourceLocationSettings,
)

TEST_APERTURE_SETTINGS = dict(radius=5, gap=10, annulus_width=15, fwhm=3.2)

TEST_CAMERA_VALUES = dict(
data_unit="adu",
gain="2.0 electron / adu",
name="test camera",
read_noise="10.0 electron",
dark_current="0.01 electron / s",
pixel_scale="0.563 arcsec / pix",
max_data_value="50000.0 adu",
)

TEST_EXOPLANET_SETTINGS = dict(
epoch={
"jd1": 0.0,
"jd2": 0.0,
"format": "jd",
"scale": "utc",
"precision": 3,
"in_subfmt": "*",
"out_subfmt": "*",
},
period="0.0 min",
identifier="a planet",
coordinate={
"ra": "0d00m00s",
"dec": "0d00m00s",
"representation_type": "spherical",
"frame": "icrs",
},
depth=0,
duration="0.0 min",
)

TEST_OBSERVATORY_SETTINGS = dict(
name="test observatory",
longitude="43d00m00s",
latitude="45d00m00s",
elevation="311.0 m",
AAVSO_code="test",
TESS_telescope_code="tess test",
)

# The first setting here is required, the rest are optional. The optional
# settings below are different than the defaults in the model definition.
TEST_PHOTOMETRY_OPTIONS = dict(
include_dig_noise=False,
reject_too_close=False,
reject_background_outliers=False,
fwhm_by_fit=False,
method="center",
)

TEST_PASSBAND_MAP = dict(
name="Example map",
your_filter_names_to_aavso=[
dict(
your_filter_name="V",
aavso_filter_name="V",
),
dict(
your_filter_name="B",
aavso_filter_name="B",
),
dict(
your_filter_name="rp",
aavso_filter_name="SR",
),
],
)

TEST_LOGGING_SETTINGS = dict(
logfile="test.log",
console_log=False,
)

TEST_SOURCE_LOCATION_SETTINGS = dict(
shift_tolerance=5,
source_list_file="test.ecsv",
use_coordinates="pixel",
)

TEST_PHOTOMETRY_SETTINGS = dict(
camera=TEST_CAMERA_VALUES,
observatory=TEST_OBSERVATORY_SETTINGS,
photometry_apertures=TEST_APERTURE_SETTINGS,
source_location_settings=TEST_SOURCE_LOCATION_SETTINGS,
photometry_optional_settings=TEST_PHOTOMETRY_OPTIONS,
passband_map=TEST_PASSBAND_MAP,
logging_settings=TEST_LOGGING_SETTINGS,
)


@pytest.mark.parametrize(
"model,settings",
Expand Down
2 changes: 1 addition & 1 deletion stellarphot/settings/tests/test_photometry_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
PhotometryWorkingDirSettings,
settings_files,
)
from stellarphot.settings.constants import TEST_PHOTOMETRY_SETTINGS
from stellarphot.settings.custom_widgets import PhotometryRunner
from stellarphot.settings.tests.test_models import TEST_PHOTOMETRY_SETTINGS


# See test_settings_file.TestSavedSettings for a detailed description of what the
Expand Down
2 changes: 1 addition & 1 deletion stellarphot/settings/tests/test_settings_file.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
SavedSettings,
settings_files, # This import is needed for mocking -- see TestSavedSettings
)
from stellarphot.settings.tests.test_models import TEST_PHOTOMETRY_SETTINGS
from stellarphot.settings.constants import TEST_PHOTOMETRY_SETTINGS

CAMERA = """
{
Expand Down
2 changes: 1 addition & 1 deletion stellarphot/settings/tests/test_views.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from stellarphot.settings import Camera, SourceLocationSettings, ui_generator
from stellarphot.settings.tests.test_models import TEST_CAMERA_VALUES
from stellarphot.settings.constants import TEST_CAMERA_VALUES


class TestUiGenerator:
Expand Down