Fix change to test

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

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -43,7 +43,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":