Call the io retrieve_alerts rather than its Schema method shadow

lsst · Jul 7, 2020 · dc6f608 · dc6f608
1 parent 207f135
commit dc6f608
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_io.py b/test/test_io.py
@@ -71,7 +71,7 @@ def test_retrieve_alerts(self):
         records = self._mock_records(5)
 
         with self._temp_alert_file(records) as alert_file:
-            have_schema, have_records_iterable = self.test_schema.retrieve_alerts(alert_file)
+            have_schema, have_records_iterable = retrieve_alerts(alert_file, self.test_schema)
             have_records = list(have_records_iterable)
 
         self.assertEqual(records, have_records)
@@ -82,7 +82,7 @@ def test_alert_file_with_one_alert(self):
         records = self._mock_records(1)
 
         with self._temp_alert_file(records) as alert_file:
-            have_schema, have_records_iterable = self.test_schema.retrieve_alerts(alert_file)
+            have_schema, have_records_iterable = retrieve_alerts(alert_file, self.test_schema)
             have_records = list(have_records_iterable)
 
         self.assertEqual(records, list(have_records))
@@ -93,7 +93,7 @@ def test_alert_file_with_no_alerts(self):
         records = []
 
         with self._temp_alert_file(records) as alert_file:
-            have_schema, have_records_iterable = self.test_schema.retrieve_alerts(alert_file)
+            have_schema, have_records_iterable = retrieve_alerts(alert_file, self.test_schema)
             have_records = list(have_records_iterable)
 
         self.assertEqual(records, list(have_records))