Address feedback

- Adjust test cases
- change valid statuses variable from list to set

Author: Rodion Yaryy

labelbox/schema/project.py

@@ -194,7 +194,7 @@ def export_issues(self, status=None):
             }
         }""" % (id_param, status_param, id_param, status_param)
 
-        valid_statuses = [None, "Open", "Resolved"]
+        valid_statuses = {None, "Open", "Resolved"}
 
         if status not in valid_statuses:
             raise ValueError("status must be in {}. Found {}".format(

tests/integration/test_issues.py

@@ -3,23 +3,18 @@
 
 def test_issues_export(project):
     exported_issues_url = project.export_issues()
-    assert exported_issues_url is not None
-    assert exported_issues_url is not ''
+    assert exported_issues_url
 
     exported_issues_url = project.export_issues("Open")
-    assert exported_issues_url is not None
-    assert exported_issues_url is not ''
+    assert exported_issues_url
     assert "?status=Open" in exported_issues_url
 
     exported_issues_url = project.export_issues("Resolved")
-    assert exported_issues_url is not None
-    assert exported_issues_url is not ''
+    assert exported_issues_url
     assert "?status=Resolved" in exported_issues_url
 
     invalidStatusValue = "Closed"
     with pytest.raises(ValueError) as exc_info:
         exported_issues_url = project.export_issues(invalidStatusValue)
-    assert str(
-        exc_info.value
-    ) == "status must be in [None, 'Open', 'Resolved']. Found {}".format(
-        invalidStatusValue)
+    assert "status must be in" in str(exc_info.value)
+    assert "Found %s" % (invalidStatusValue) in str(exc_info.value)