barakplasma commited on
Commit
d484982
·
1 Parent(s): f572466
Files changed (1) hide show
  1. test_app.py +13 -13
test_app.py CHANGED
@@ -10,26 +10,26 @@ def test_parse_requirements_with_spaces() -> None:
10
  ]
11
  dependencies, all_tasks, original_names = parse_requirements(reqs)
12
  assert set(all_tasks) == {
13
- "take out trash",
14
  "bag",
15
  "dinner",
16
  "shopping",
17
- "make sandwich",
18
  "bread",
19
  }
20
- assert dependencies["take out trash"] == ["bag"]
21
  assert dependencies["dinner"] == ["shopping"]
22
- assert dependencies["make sandwich"] == ["bread"]
23
- assert original_names["take out trash"] == "take out trash"
24
 
25
 
26
  def test_parse_tasks_and_original_names() -> None:
27
  txt = "Wash dishes\nTake out trash, Make sandwich "
28
  tasks, originals = parse_tasks(txt)
29
- assert set(tasks) == {"wash dishes", "take out trash", "make sandwich"}
30
- assert originals["take out trash"] == "Take out trash"
31
- assert originals["make sandwich"] == "Make sandwich"
32
- assert originals["wash dishes"] == "Wash dishes"
33
 
34
 
35
  def test_solve_all_tasks_simple() -> None:
@@ -43,7 +43,7 @@ def test_solve_all_tasks_with_spaces() -> None:
43
  reqs = ["Do homework requires eat dinner", "eat dinner requires cook"]
44
  dependencies, all_tasks, _ = parse_requirements(reqs)
45
  result = solve_all_tasks(dependencies, all_tasks)
46
- assert result == ["cook", "eat dinner", "do homework"]
47
 
48
 
49
  # def test_solve_maximum_subset_cycle():
@@ -61,6 +61,6 @@ def test_generate_mermaid_gantt_minutes() -> None:
61
  base = datetime(2024, 1, 1, 8, 0)
62
  gantt = generate_mermaid_gantt(order, names, durations=durations, start_time=base)
63
  assert "dateFormat YYYY-MM-DD HH:mm" in gantt
64
- assert ": 2024-01-01 08:00, 15m" in gantt
65
- assert ": 2024-01-01 08:15, 30m" in gantt
66
- assert ": 2024-01-01 08:45, 45m" in gantt
 
10
  ]
11
  dependencies, all_tasks, original_names = parse_requirements(reqs)
12
  assert set(all_tasks) == {
13
+ "take_out_trash",
14
  "bag",
15
  "dinner",
16
  "shopping",
17
+ "make_sandwich",
18
  "bread",
19
  }
20
+ assert dependencies["take_out_trash"] == ["bag"]
21
  assert dependencies["dinner"] == ["shopping"]
22
+ assert dependencies["make_sandwich"] == ["bread"]
23
+ assert original_names["take_out_trash"] == "take out trash"
24
 
25
 
26
  def test_parse_tasks_and_original_names() -> None:
27
  txt = "Wash dishes\nTake out trash, Make sandwich "
28
  tasks, originals = parse_tasks(txt)
29
+ assert set(tasks) == {"wash_dishes", "take_out_trash", "make_sandwich"}
30
+ assert originals["take_out_trash"] == "Take out trash"
31
+ assert originals["make_sandwich"] == "Make sandwich"
32
+ assert originals["wash_dishes"] == "Wash dishes"
33
 
34
 
35
  def test_solve_all_tasks_simple() -> None:
 
43
  reqs = ["Do homework requires eat dinner", "eat dinner requires cook"]
44
  dependencies, all_tasks, _ = parse_requirements(reqs)
45
  result = solve_all_tasks(dependencies, all_tasks)
46
+ assert result == ["cook", "eat_dinner", "do_homework"]
47
 
48
 
49
  # def test_solve_maximum_subset_cycle():
 
61
  base = datetime(2024, 1, 1, 8, 0)
62
  gantt = generate_mermaid_gantt(order, names, durations=durations, start_time=base)
63
  assert "dateFormat YYYY-MM-DD HH:mm" in gantt
64
+ assert "A : a, 2024-01-01 08:00, 15m" in gantt
65
+ assert "B : b, after a, 30m" in gantt
66
+ assert "C : c, after b, 45m" in gantt