ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

WIP: PYTHON IS TRASH

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-03-05 13:57:03 -05:00
parent 5e69617f88
commit 01c5bb7947
5 changed files with 341 additions and 434 deletions
Download Patch File Download Diff File Expand all files Collapse all files

126
tests/test_dag_engine.py
View File

@@ -1,73 +1,73 @@
import pytest
from models import Ticket
from dag_engine import TrackDAG
from src.models import Ticket
from src.dag_engine import TrackDAG
def test_get_ready_tasks_linear() -> None:
t1 = Ticket(id="T1", description="Task 1", status="completed", assigned_to="worker")
t2 = Ticket(id="T2", description="Task 2", status="todo", assigned_to="worker", depends_on=["T1"])
t3 = Ticket(id="T3", description="Task 3", status="todo", assigned_to="worker", depends_on=["T2"])
def test_get_ready_tasks_linear():
t1 = Ticket(id="T1", description="desc", status="todo")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
dag = TrackDAG([t1, t2])
ready = dag.get_ready_tasks()
assert len(ready) == 1
assert ready[0].id == "T1"
def test_get_ready_tasks_branching():
t1 = Ticket(id="T1", description="desc", status="completed")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
t3 = Ticket(id="T3", description="desc", status="todo", depends_on=["T1"])
dag = TrackDAG([t1, t2, t3])
ready = dag.get_ready_tasks()
assert len(ready) == 2
ids = [t.id for t in ready]
assert "T2" in ids
assert "T3" in ids
def test_has_cycle_no_cycle():
t1 = Ticket(id="T1", description="desc", status="todo")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
dag = TrackDAG([t1, t2])
assert dag.has_cycle() is False
def test_has_cycle_direct_cycle():
t1 = Ticket(id="T1", description="desc", status="todo", depends_on=["T2"])
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
dag = TrackDAG([t1, t2])
assert dag.has_cycle() is True
def test_has_cycle_indirect_cycle():
t1 = Ticket(id="T1", description="desc", status="todo", depends_on=["T3"])
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
t3 = Ticket(id="T3", description="desc", status="todo", depends_on=["T2"])
dag = TrackDAG([t1, t2, t3])
assert dag.has_cycle() is True
def test_has_cycle_complex_no_cycle():
t1 = Ticket(id="T1", description="desc", status="todo")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
t3 = Ticket(id="T3", description="desc", status="todo", depends_on=["T1"])
t4 = Ticket(id="T4", description="desc", status="todo", depends_on=["T2", "T3"])
dag = TrackDAG([t1, t2, t3, t4])
assert dag.has_cycle() is False
def test_get_ready_tasks_multiple_deps():
t1 = Ticket(id="T1", description="desc", status="completed")
t2 = Ticket(id="T2", description="desc", status="todo")
t3 = Ticket(id="T3", description="desc", status="todo", depends_on=["T1", "T2"])
dag = TrackDAG([t1, t2, t3])
# Only T2 is ready because T3 depends on T2 (todo)
ready = dag.get_ready_tasks()
assert len(ready) == 1
assert ready[0].id == "T2"
def test_get_ready_tasks_branching() -> None:
t1 = Ticket(id="T1", description="Task 1", status="completed", assigned_to="worker")
t2 = Ticket(id="T2", description="Task 2", status="todo", assigned_to="worker", depends_on=["T1"])
t3 = Ticket(id="T3", description="Task 3", status="todo", assigned_to="worker", depends_on=["T1"])
dag = TrackDAG([t1, t2, t3])
ready = dag.get_ready_tasks()
assert len(ready) == 2
ready_ids = {t.id for t in ready}
assert ready_ids == {"T2", "T3"}
def test_topological_sort():
t1 = Ticket(id="T1", description="desc", status="todo")
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
dag = TrackDAG([t2, t1]) # Out of order input
sorted_tasks = dag.topological_sort()
assert [t.id for t in sorted_tasks] == ["T1", "T2"]
def test_has_cycle_no_cycle() -> None:
t1 = Ticket(id="T1", description="Task 1", status="todo", assigned_to="worker")
t2 = Ticket(id="T2", description="Task 2", status="todo", assigned_to="worker", depends_on=["T1"])
def test_topological_sort_cycle():
t1 = Ticket(id="T1", description="desc", status="todo", depends_on=["T2"])
t2 = Ticket(id="T2", description="desc", status="todo", depends_on=["T1"])
dag = TrackDAG([t1, t2])
assert not dag.has_cycle()
def test_has_cycle_direct_cycle() -> None:
t1 = Ticket(id="T1", description="Task 1", status="todo", assigned_to="worker", depends_on=["T2"])
t2 = Ticket(id="T2", description="Task 2", status="todo", assigned_to="worker", depends_on=["T1"])
dag = TrackDAG([t1, t2])
assert dag.has_cycle()
def test_has_cycle_indirect_cycle() -> None:
t1 = Ticket(id="T1", description="Task 1", status="todo", assigned_to="worker", depends_on=["T2"])
t2 = Ticket(id="T2", description="Task 2", status="todo", assigned_to="worker", depends_on=["T3"])
t3 = Ticket(id="T3", description="Task 3", status="todo", assigned_to="worker", depends_on=["T1"])
dag = TrackDAG([t1, t2, t3])
assert dag.has_cycle()
def test_has_cycle_complex_no_cycle() -> None:
t1 = Ticket(id="T1", description="T1", status="todo", assigned_to="worker", depends_on=["T2", "T3"])
t2 = Ticket(id="T2", description="T2", status="todo", assigned_to="worker", depends_on=["T4"])
t3 = Ticket(id="T3", description="T3", status="todo", assigned_to="worker", depends_on=["T4"])
t4 = Ticket(id="T4", description="T4", status="todo", assigned_to="worker")
dag = TrackDAG([t1, t2, t3, t4])
assert not dag.has_cycle()
def test_get_ready_tasks_multiple_deps() -> None:
t1 = Ticket(id="T1", description="T1", status="completed", assigned_to="worker")
t2 = Ticket(id="T2", description="T2", status="completed", assigned_to="worker")
t3 = Ticket(id="T3", description="T3", status="todo", assigned_to="worker", depends_on=["T1", "T2"])
dag = TrackDAG([t1, t2, t3])
assert [t.id for t in dag.get_ready_tasks()] == ["T3"]
t2.status = "todo"
assert [t.id for t in dag.get_ready_tasks()] == ["T2"]
def test_topological_sort() -> None:
t1 = Ticket(id="T1", description="T1", status="todo", assigned_to="worker")
t2 = Ticket(id="T2", description="T2", status="todo", assigned_to="worker", depends_on=["T1"])
t3 = Ticket(id="T3", description="T3", status="todo", assigned_to="worker", depends_on=["T2"])
dag = TrackDAG([t1, t2, t3])
sort = dag.topological_sort()
assert sort == ["T1", "T2", "T3"]
def test_topological_sort_cycle() -> None:
t1 = Ticket(id="T1", description="T1", status="todo", assigned_to="worker", depends_on=["T2"])
t2 = Ticket(id="T2", description="T2", status="todo", assigned_to="worker", depends_on=["T1"])
dag = TrackDAG([t1, t2])
with pytest.raises(ValueError, match="Dependency cycle detected"):
with pytest.raises(ValueError, match="DAG Validation Error: Cycle detected"):
dag.topological_sort()