Fix change to test

drivendataorg · Jul 2, 2020 · 1412c15 · 1412c15
1 parent 5095d2e
commit 1412c15
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -31,7 +31,12 @@ def test_version():
 
 
 def test_main_python_m():
-    result = subprocess.run(["python", "-m", "nbautoexport"], capture_output=True, text=True)
+    result = subprocess.run(
+        ["python", "-m", "nbautoexport"],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        universal_newlines=True,
+    )
     assert result.returncode == 0
     assert "Exports Jupyter notebooks to various file formats" in result.stdout
     assert result.stdout.startswith("Usage: python -m nbautoexport")
@@ -43,7 +48,7 @@ def test_version_python_m():
         ["python", "-m", "nbautoexport", "--version"],
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE,
-        text=True,
+        universal_newlines=True,
     )
     assert result.returncode == 0
     assert result.stdout.strip() == __version__
diff --git a/tests/test_export.py b/tests/test_export.py
@@ -28,7 +28,7 @@ def test_export_notebook(notebooks_dir, organize_by):
     export_notebook(notebook.path, config)
 
     expected_exports = set()
-    for fmt in ExportFormat:
+    for fmt in EXPORT_FORMATS_TO_TEST:
         if organize_by == "extension":
             subfolder = notebooks_dir / fmt.value
         elif organize_by == "notebook":