Merge pull request #2649 from ziv17/2596-fix-set-critical

Fixed NewsFlash.set_critical()
data-for-change · May 14, 2024 · bdee997 · bdee997
2 parents 8a94090 + 51c45f2
commit bdee997
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/anyway/models.py b/anyway/models.py
@@ -903,7 +903,7 @@ def set_critical(
         resolution = BE_CONST.ResolutionCategories(self.resolution)
         end_time = last_accident_date.to_pydatetime().date()
         start_time = datetime.date(end_time.year + 1 - years_before, 1, 1)
-        location_info = LocationInfo()
+        location_info: LocationInfo = {}
         if resolution == BE_CONST.ResolutionCategories.SUBURBAN_ROAD:
             location_info["road1"] = self.road1
             location_info["road_segment_id"] = self.road_segment_id

diff --git a/tests/test_news_flash.py b/tests/test_news_flash.py
@@ -144,6 +144,7 @@ def test_twitter_parse():
         for k in raw_fields:
             assert getattr(actual, k) == getattr(expected, k)
 
+        actual.set_critical()
         actual.accident = classify_tweets(actual.description)
         assert actual.accident == expected.accident