Fixing unit test since we no longer filter out nonrev trips

transitmatters · Nov 27, 2024 · bed9dc9 · bed9dc9
1 parent 284f998
commit bed9dc9
Show file tree

Hide file tree

Showing 3 changed files with 596 additions and 539 deletions.
diff --git a/mbta-performance/chalicelib/lamp/tests/test_ingest.py b/mbta-performance/chalicelib/lamp/tests/test_ingest.py
@@ -99,7 +99,7 @@ def test_ingest_pq_file(self):
         null_id_events = pq_df_after[pq_df_after["stop_id"].isna()]
         self.assertEqual(added.shape, (3763, 17))
         self.assertTrue(null_id_events.empty)
-        self.assertEqual(pq_df_after.shape, (16700, 17))
+        self.assertEqual(pq_df_after.shape, (17074, 17))
         self.assertEqual(set(pq_df_after["service_date"].unique()), {"2024-02-07"})
 
     def test__average_scheduled_headways(self):