Include due dates in projects

Author: leonlan

psplib/ProjectInstance.py

@@ -94,10 +94,13 @@ class Project:
         The activities indices that belong to this project.
     release_date
         The earliest start time of this project.
+    due_date
+        The due date of this project, if available. Default is ``None``.
     """
 
     activities: list[int]
     release_date: int = 0
+    due_date: Optional[int] = None
 
     @property
     def num_activities(self):

psplib/parse_mplib.py

@@ -33,11 +33,20 @@ def parse_mplib(loc: Union[str, Path]) -> ProjectInstance:
     id2idx: dict[str, int] = {}  # maps activity names to idcs
 
     for project_idx in range(1, num_projects + 1):
-        num_activities, release_date = map(int, next(lines).split())
+        project_data = list(map(int, next(lines).split()))
+
+        if len(project_data) == 3:
+            # Modified MPLIB instances that include due date.
+            num_activities, release_date, due_date = project_data
+        else:
+            # Original MPLIB instances do not include due date.
+            num_activities, release_date = project_data
+            due_date = None
+
         next(lines)  # denotes used resources, implies that demand > 0
 
         idcs = [len(activities) + idx for idx in range(num_activities)]
-        projects.append(Project(idcs, release_date))
+        projects.append(Project(idcs, release_date, due_date))
 
         for activity_idx in range(1, num_activities + 1):
             line = next(lines).split()

tests/test_parse_mplib.py

@@ -34,6 +34,7 @@ def test_mplib_set1():
     for project in instance.projects:
         assert_equal(project.num_activities, 62)
         assert_equal(project.release_date, 0)
+        assert_equal(project.due_date, None)
 
 
 def test_mplib_set2():
@@ -66,3 +67,21 @@ def test_mplib_set2():
     for project in instance.projects:
         assert_equal(project.num_activities, 52)
         assert_equal(project.release_date, 0)
+        assert_equal(project.due_date, None)
+
+
+def test_mplib_set1_with_due_date():
+    """
+    Tests that the instance ``MPLIB1_Set1_0_due_date.rcmp`` is correctly
+    parsed, which includes due dates for each project.
+    """
+    instance = parse_mplib(relative("data/MPLIB1_Set1_0_due_date.rcmp"))
+
+    # It's the same as the original instance, but now due dates.
+    due_dates = [113, 96, 117, 138, 216, 233]
+
+    assert_equal(instance.num_projects, 6)
+    for idx, project in enumerate(instance.projects):
+        assert_equal(project.num_activities, 62)
+        assert_equal(project.release_date, 0)
+        assert_equal(project.due_date, due_dates[idx])