diff --git a/openpilot/starpilot/galaxy/tests/test_recording_library.py b/openpilot/starpilot/galaxy/tests/test_recording_library.py index 5791e0a3b9..dbe1625c06 100644 --- a/openpilot/starpilot/galaxy/tests/test_recording_library.py +++ b/openpilot/starpilot/galaxy/tests/test_recording_library.py @@ -77,6 +77,8 @@ class RecordingLibraryTest(unittest.TestCase): self.assertFalse(self.segment.exists()) def test_selected_route_inventory_and_archive_are_complete_beyond_list_limit(self): + self.history = DriveHistory(self.root, max_segments_per_route=64, max_segments=64) + self.library = RecordingLibrary(self.history) for number in range(1, 70): folder = self.root / f'{ROUTE}--{number}' folder.mkdir() diff --git a/openpilot/starpilot/replay/tests/test_catalog.py b/openpilot/starpilot/replay/tests/test_catalog.py index 7bf109390f..85c65e8644 100644 --- a/openpilot/starpilot/replay/tests/test_catalog.py +++ b/openpilot/starpilot/replay/tests/test_catalog.py @@ -105,7 +105,8 @@ class RouteInventoryTest(unittest.TestCase): def test_long_selected_route_and_unrelated_inventory_limits(self): for number in range(65): self.segment(number) - self.assertTrue(DriveHistory(self.root).snapshot()['scanIncomplete']) + history = DriveHistory(self.root, max_segments_per_route=64, max_segments=64) + self.assertTrue(history.snapshot()['scanIncomplete']) self.assertEqual(self.catalog.detail(ROUTE)['segmentCount'], 65) with self.catalog.lease(ROUTE) as lease: self.assertTrue(lease.current())