Added regression tests for issue ContinualAI#774

lrzpellegrini · May 30, 2023 · eef6333 · eef6333
1 parent ce12590
commit eef6333
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_core50.py b/tests/test_core50.py
@@ -38,6 +38,17 @@ def test_core50_nc_benchmark(self):
         classes_in_test = benchmark_instance.classes_in_experience["test"][0]
         self.assertSetEqual(set(range(50)), set(classes_in_test))
 
+        # Regression tests for issue #774
+        self.assertSequenceEqual(
+            [10] + ([5] * 8),
+            benchmark_instance.n_classes_per_exp)
+        self.assertSetEqual(
+            set(range(50)),
+            set(benchmark_instance.classes_order))
+        self.assertEqual(
+            50,
+            len(benchmark_instance.classes_order))
+
 
 if __name__ == "__main__":
     unittest.main()