Python 2 syntax fix

materials-data-facility · Jan 5, 2018 · 39381ba · 39381ba
1 parent d44cdbf
commit 39381ba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/mdf_forge/forge.py b/mdf_forge/forge.py
@@ -935,10 +935,11 @@ def http_stream(self, results, verbose=True):
         """
         if self.__anonymous:
             print_("Error: Anonymous HTTP download not yet supported.")
-            return {
+            yield {
                 "success": False,
                 "message": "Anonymous HTTP download not yet supported."
                 }
+            return
         # If results have info attached, remove it
         if type(results) is tuple:
             results = results[0]

diff --git a/tests/test_forge.py b/tests/test_forge.py
@@ -889,7 +889,9 @@ def test_forge_anonymous(capsys):
     out, err = capsys.readouterr()
     assert "Error: Anonymous Globus Transfer not supported." in out
     # http_stream
-    with pytest.raises(StopIteration):
-        next(f.http_stream({}))
+    res = f.http_stream({})
+    assert next(res)["success"] is False
     out, err = capsys.readouterr()
     assert "Error: Anonymous HTTP download not yet supported." in out
+    with pytest.raises(StopIteration):
+        next(res)