feat(env): Course Schedule Curriculum (#266)

* course schedule curriculum

* update levels

* update comments

* lint

tests/test_course_schedule.py

@@ -2,31 +2,31 @@
 
 import pytest
 
-from reasoning_gym.graphs.course_schedule import CourseScheduleConfig, CourseScheduleDataset
+from reasoning_gym.graphs.course_schedule import CourseScheduleConfig, CourseScheduleCurriculum, CourseScheduleDataset
 
 
 def test_course_schedule_config_validation():
     """Test that invalid configs raise appropriate errors"""
     with pytest.raises(AssertionError):
-        config = CourseScheduleConfig(num_courses=-1)  # Negative not allowed
+        config = CourseScheduleConfig(min_num_courses=2)  # must be >= 3
         config.validate()
 
     with pytest.raises(AssertionError):
-        config = CourseScheduleConfig(num_courses=0)  # Zero not allowed
+        config = CourseScheduleConfig(min_num_courses=6, max_num_courses=5)  # min > max
         config.validate()
 
     with pytest.raises(AssertionError):
-        config = CourseScheduleConfig(max_num_prerequisites=-1)  # Negative not allowed
+        config = CourseScheduleConfig(min_num_prerequisites=-1)  # neg not allowed
+        config.validate()
+
+    with pytest.raises(AssertionError):
+        config = CourseScheduleConfig(min_num_prerequisites=5, max_num_prerequisites=4)  # min > max
         config.validate()
 
     with pytest.raises(AssertionError):
         config = CourseScheduleConfig(max_num_prerequisites=0)  # Zero not allowed
         config.validate()
 
-    with pytest.raises(AssertionError):
-        config = CourseScheduleConfig(num_courses=3, max_num_prerequisites=5)  # max_num_prerequisites > num_courses
-        config.validate()
-
     with pytest.raises(AssertionError):
         config = CourseScheduleConfig(p_solvable=-0.1)  # < 0 not allowed
         config.validate()
@@ -60,7 +60,7 @@ def test_course_schedule_dataset_deterministic():
 
 def test_course_schedule_dataset_items():
     """Test basic properties of generated items"""
-    config = CourseScheduleConfig(num_courses=15, size=10, seed=42)
+    config = CourseScheduleConfig(max_num_courses=15, size=10, seed=42)
     dataset = CourseScheduleDataset(config)
 
     for i in range(len(dataset)):
@@ -83,13 +83,12 @@ def test_course_schedule_dataset_items():
         solution = item["metadata"]["solution"]  # Solution obtained from topological sort
 
         # Verify metadata
-        assert len(courses) == config.num_courses
-        assert max(courses) == config.num_courses - 1
-        assert len(prerequisites) <= config.max_num_prerequisites * config.num_courses
+        assert len(courses) <= config.max_num_courses
+        assert len(prerequisites) <= config.max_num_prerequisites * len(courses)
         assert all(len(prereq) == 2 for prereq in prerequisites)
         for course, prereq in prerequisites:
-            assert course < config.num_courses
-            assert prereq < config.num_courses
+            assert course < len(courses)
+            assert prereq < len(courses)
             assert course != prereq
         assert solution == solvable
 
@@ -125,3 +124,32 @@ def test_course_schedule_answer():
     # Indirect cycle of length 3
     prerequisites = [[0, 1], [1, 2], [2, 0]]
     assert dataset._can_finish(num_courses=3, prerequisites=prerequisites) == False
+
+
+def test_course_schedule_curriculum():
+    curriculum = CourseScheduleCurriculum()
+
+    base_value = {"size": 150, "seed": 1}
+
+    base_cfg: CourseScheduleConfig = curriculum.generate_configuration(base_value)
+    assert base_cfg.seed == 1
+    assert base_cfg.size == 150
+    assert base_cfg.min_num_courses == 10 and base_cfg.max_num_courses == 10
+    assert base_cfg.min_num_prerequisites == 2 and base_cfg.max_num_prerequisites == 2
+    assert base_cfg.min_cycle_length == 3 and base_cfg.max_cycle_length == 3
+
+    # test incrementing attribute levels
+    curriculum.increment_attr_level("num_courses")
+    curriculum.increment_attr_level("num_prerequisites")
+    curriculum.increment_attr_level("cycle_length")
+    increased_cfg = curriculum.generate_configuration(base_value)
+    assert increased_cfg.min_num_courses == 10 and increased_cfg.max_num_courses == 50
+    assert increased_cfg.min_num_prerequisites == 2 and increased_cfg.max_num_prerequisites == 3
+    assert increased_cfg.min_cycle_length == 3 and increased_cfg.max_cycle_length == 4
+
+    # test decrementing attribute level for num_courses again
+    curriculum.decrement_attr_level("num_courses")
+    partially_decreased_cfg = curriculum.generate_configuration(base_value)
+    assert partially_decreased_cfg.min_num_courses == 10 and partially_decreased_cfg.max_num_courses == 10
+    assert partially_decreased_cfg.min_num_prerequisites == 2 and partially_decreased_cfg.max_num_prerequisites == 3
+    assert partially_decreased_cfg.min_cycle_length == 3 and partially_decreased_cfg.max_cycle_length == 4