Improve coverage

datasciencecampus · Oct 30, 2023 · 539ff66 · 539ff66
1 parent eb1e7c3
commit 539ff66
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/gtfs/test_cleaners.py b/tests/gtfs/test_cleaners.py
@@ -14,7 +14,10 @@
     clean_unrecognised_column_warnings,
     clean_duplicate_stop_times,
 )
-from transport_performance.gtfs.gtfs_utils import _get_validation_warnings
+from transport_performance.gtfs.gtfs_utils import (
+    _get_validation_warnings,
+    _remove_validation_row,
+)
 
 
 @pytest.fixture(scope="function")
@@ -468,3 +471,7 @@ def test_clean_duplicate_stop_times_on_pass(self):
         assert (
             len(gtfs.feed.stop_times.loc[18]) > 0
         ), "Test case removed from data"
+
+        # test return of none when no warnings are found (for cov)
+        _remove_validation_row(gtfs, message=r".* \(trip_id, departure_time\)")
+        assert clean_duplicate_stop_times(gtfs) is None