Skip to content

Commit

Permalink
Working on tests
Browse files Browse the repository at this point in the history
  • Loading branch information
micah-prime committed Oct 3, 2024
1 parent a144963 commit 07c1bc7
Showing 1 changed file with 16 additions and 11 deletions.
27 changes: 16 additions & 11 deletions tests/db_connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,13 +63,25 @@ def _add_entry(
session, Campaign, dict(name=campaign_name)
)

# add list of observers
observer_list = []
for obs_name in observer_names:
observer = cls._check_or_add_object(
session, Observer, dict(name=obs_name)
)
observer_list.append(observer)

# Add site
if site_name is not None:
site = cls._check_or_add_object(
session, Site, dict(name=site_name),
object_kwargs=dict(
name=site_name, campaign=campaign,
date=kwargs.pop("date")
date=kwargs.pop("date"),
geom=kwargs.pop("geom"),
time=kwargs.pop("time"),
elevation=kwargs.pop("elevation"),
observers=observer_list,
)
)
else:
Expand All @@ -85,21 +97,14 @@ def _add_entry(
session, MeasurementType, dict(name=measurement_type)
)

# add list of observers
observer_list = []
for obs_name in observer_names:
observer = cls._check_or_add_object(
session, Observer, dict(name=obs_name)
)
observer_list.append(observer)

object_kwargs = dict(
instrument=instrument, observers=observer_list,
doi=doi, measurement=measurement_obj, **kwargs
instrument=instrument, doi=doi, measurement=measurement_obj,
**kwargs
)
# Add site if given
if site_name is None:
object_kwargs["campaign"] = campaign
object_kwargs["observers"] = observer_list
else:
object_kwargs["site"] = site

Expand Down

0 comments on commit 07c1bc7

Please sign in to comment.