feat: add milestone to task in test_defect_106b

Author: csgoh

src/tests/test_defects.py

@@ -31,6 +31,7 @@ def test_defect_106a():
 
     roadmap.save("images/defects/issue-106.png")
 
+
 @pytest.mark.unit
 def test_defect_106b():
     roadmap = Roadmap(1800, 400, colour_theme="BLUEMOUNTAIN", show_marker=True)
@@ -47,16 +48,15 @@ def test_defect_106b():
     group = roadmap.add_group("Core Product Work Stream")
 
     group.add_task("Base Functionality", "2024-12-01", "2025-01-05")
-    group.add_task("Enhancements12", "2025-01-06", "2025-03-31")
+    g1 = group.add_task("Enhancements12", "2025-01-06", "2025-03-31")
+    g1.add_milestone("Milestone 1", "2025-02-05")
 
     roadmap.set_footer("Generated by Roadmapper")
     roadmap.draw()
 
     roadmap.save("images/defects/issue-106b.png")
 
 
-
-
 if not os.path.exists("images"):
     os.mkdir("images")