adds test again

ploomber · Oct 5, 2020 · 00c78dc · 00c78dc
1 parent db2c266
commit 00c78dc
Showing 1 changed file with 21 additions and 21 deletions.
diff --git a/tests/test_notebook_task.py b/tests/test_notebook_task.py
@@ -363,24 +363,24 @@ def test_hot_reload(tmp_directory):
     # TODO: check task is not marked as outdated
 
 
-# @pytest.mark.parametrize('kind', ['ipdb', 'pdb', 'pm'])
-# def test_debug(kind, tmp_directory):
-#     dag = DAG()
-
-#     code = """
-# # + tags=["parameters"]
-# 1 + 1
-#     """
-
-#     t = NotebookRunner(code,
-#                        product=File(Path(tmp_directory, 'out.ipynb')),
-#                        dag=dag,
-#                        kernelspec_name='python3',
-#                        params={'var': 1},
-#                        ext_in='py',
-#                        name='nb')
-
-#     dag.render()
-
-#     with mock.patch.object(builtins, 'input', lambda *args: 'quit'):
-#         t.debug(kind=kind)
+@pytest.mark.parametrize('kind', ['ipdb', 'pdb', 'pm'])
+def test_debug(kind, tmp_directory):
+    dag = DAG()
+
+    code = """
+# + tags=["parameters"]
+1 + 1
+    """
+
+    t = NotebookRunner(code,
+                       product=File(Path(tmp_directory, 'out.ipynb')),
+                       dag=dag,
+                       kernelspec_name='python3',
+                       params={'var': 1},
+                       ext_in='py',
+                       name='nb')
+
+    dag.render()
+
+    with mock.patch.object(builtins, 'input', lambda *args: 'quit'):
+        t.debug(kind=kind)