Skip to content

Commit cd39f61

Browse files
committed
added test for get_event_counts
1 parent 72746fc commit cd39f61

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/test_forecast.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,10 @@ def test_ascii_some_missing_verbose(self):
5151
self.assertEqual(10, test_fore.n_cat)
5252
numpy.testing.assert_array_equal([cat.catalog_id for cat in test_fore], numpy.arange(10))
5353

54-
def test_ascii_all_present(self):
55-
pass
54+
def test_get_event_counts(self):
55+
fname = os.path.join(get_test_catalog_root(), 'all_present.csv')
56+
test_fore = load_catalog_forecast(fname)
57+
numpy.testing.assert_array_equal(numpy.ones(10), test_fore.get_event_counts())
5658

5759
if __name__ == '__main__':
5860
unittest.main()

0 commit comments

Comments
 (0)