Add start and end time for 'Site' to factories. Make those times time…

…zone aware
SnowEx · Oct 23, 2024 · 9db76b9 · 9db76b9
1 parent a05a6c6
commit 9db76b9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/snowexsql/tables/site.py b/snowexsql/tables/site.py
@@ -53,8 +53,8 @@ class Site(SingleLocationData, Base, InCampaign, HasDOI):
     vegetation_height = Column(String())
     tree_canopy = Column(String())
     site_notes = Column(String())
-    start_time = Column(Time)
-    end_time = Column(Time)
+    start_time = Column(Time(timezone=True))
+    end_time = Column(Time(timezone=True))
 
     @hybrid_property
     def date(self):

diff --git a/tests/factories/layer_data.py b/tests/factories/layer_data.py
@@ -25,4 +25,6 @@ class Meta:
     )
     instrument = factory.SubFactory(InstrumentFactory, name='Density Cutter')
     doi = factory.SubFactory(DOIFactory)
-    site = factory.SubFactory(SiteFactory)
+    site = factory.SubFactory(
+        SiteFactory, start_time='10:32:00', end_time='10:39:00'
+    )
diff --git a/tests/factories/site.py b/tests/factories/site.py
@@ -31,6 +31,8 @@ class Meta:
     vegetation_height = "None"
     tree_canopy = "Open"
     site_notes = "Site Notes"
+    start_time = '10:32:00'
+    end_time = '10:39:00'
 
     # Single Location data
     geom = WKTElement(