Remove flaky tests.

Author: tobiasraabe

tests/test_execute.py

@@ -20,68 +20,41 @@ class Session:
 
 @pytest.mark.end_to_end()
 @pytest.mark.parametrize("parallel_backend", PARALLEL_BACKENDS)
-def test_parallel_execution_speedup(tmp_path, parallel_backend):
+def test_parallel_execution(tmp_path, parallel_backend):
     source = """
     import pytask
-    import time
 
     @pytask.mark.produces("out_1.txt")
     def task_1(produces):
-        time.sleep(5)
         produces.write_text("1")
 
     @pytask.mark.produces("out_2.txt")
     def task_2(produces):
-        time.sleep(5)
         produces.write_text("2")
     """
     tmp_path.joinpath("task_example.py").write_text(textwrap.dedent(source))
-
-    with restore_sys_path_and_module_after_test_execution():
-        session = build(paths=tmp_path)
-
-    assert session.exit_code == ExitCode.OK
-    assert session.execution_end - session.execution_start > 10
-
-    tmp_path.joinpath("out_1.txt").unlink()
-    tmp_path.joinpath("out_2.txt").unlink()
-
     session = build(paths=tmp_path, n_workers=2, parallel_backend=parallel_backend)
-
     assert session.exit_code == ExitCode.OK
-    assert session.execution_end - session.execution_start < 10
+    assert len(session.tasks) == 2
+    assert tmp_path.joinpath("out_1.txt").exists()
+    assert tmp_path.joinpath("out_2.txt").exists()
 
 
 @pytest.mark.end_to_end()
 @pytest.mark.parametrize("parallel_backend", PARALLEL_BACKENDS)
-def test_parallel_execution_speedup_w_cli(runner, tmp_path, parallel_backend):
+def test_parallel_execution_w_cli(runner, tmp_path, parallel_backend):
     source = """
     import pytask
-    import time
 
     @pytask.mark.produces("out_1.txt")
     def task_1(produces):
-        time.sleep(5)
         produces.write_text("1")
 
     @pytask.mark.produces("out_2.txt")
     def task_2(produces):
-        time.sleep(5)
         produces.write_text("2")
     """
     tmp_path.joinpath("task_example.py").write_text(textwrap.dedent(source))
-
-    start = time()
-    result = runner.invoke(cli, [tmp_path.as_posix()])
-    end = time()
-
-    assert result.exit_code == ExitCode.OK
-    assert end - start > 10
-
-    tmp_path.joinpath("out_1.txt").unlink()
-    tmp_path.joinpath("out_2.txt").unlink()
-
-    start = time()
     result = runner.invoke(
         cli,
         [
@@ -92,11 +65,9 @@ def task_2(produces):
             parallel_backend,
         ],
     )
-    end = time()
-
     assert result.exit_code == ExitCode.OK
-    assert "Started 2 workers." in result.output
-    assert end - start < 10
+    assert tmp_path.joinpath("out_1.txt").exists()
+    assert tmp_path.joinpath("out_2.txt").exists()
 
 
 @pytest.mark.end_to_end()