handle case where the end date of a historical storm is NaT

Author: zacharyburnett

stormevents/stormevent.py

@@ -261,7 +261,10 @@ def status(self) -> StormStatus:
         entry = self.__entry
         age = datetime.today() - entry["end_date"]
         if pandas.isna(entry["end_date"]) or age < timedelta(days=1):
-            return StormStatus.REALTIME
+            if datetime.today() - entry["start_date"] > timedelta(days=30):
+                return StormStatus.HISTORICAL
+            else:
+                return StormStatus.REALTIME
         else:
             return StormStatus.HISTORICAL
 

tests/test_stormevent.py

@@ -233,6 +233,9 @@ def test_status():
     latest_storm = StormEvent.from_nhc_code(latest_storm_entry.name)
     age = datetime.today() - latest_storm_entry["end_date"]
     if pandas.isna(latest_storm_entry["end_date"]) or age < timedelta(days=1):
-        assert latest_storm.status == StormStatus.REALTIME
+        if datetime.today() - latest_storm_entry["start_date"] > timedelta(days=30):
+            assert latest_storm.status == StormStatus.HISTORICAL
+        else:
+            assert latest_storm.status == StormStatus.REALTIME
     else:
         assert latest_storm.status == StormStatus.HISTORICAL