Make dataset tests parallelizable

Author: whistler

tests/integration/test_dataset.py

@@ -7,27 +7,19 @@
 
 
 def test_dataset(client, rand_gen):
-    before = list(client.get_datasets())
-    for o in before:
-        assert isinstance(o, Dataset)
 
+    # confirm dataset can be created
     name = rand_gen(str)
     dataset = client.create_dataset(name=name)
     assert dataset.name == name
     assert dataset.created_by() == client.get_user()
     assert dataset.organization() == client.get_organization()
 
-    after = list(client.get_datasets())
-    assert len(after) == len(before) + 1
-    assert dataset in after
-
-    # confirm get_one returns first dataset
-    get_one_dataset = client.get_datasets().get_one()
-    assert get_one_dataset.uid == after[0].uid
-
-    # confirm get_many(1) returns first dataset
-    get_many_datasets = client.get_datasets().get_many(1)
-    assert get_many_datasets[0].uid == after[0].uid
+    retrieved_dataset = client.get_dataset(dataset.uid)
+    assert retrieved_dataset.name == dataset.name
+    assert retrieved_dataset.uid == dataset.uid
+    assert retrieved_dataset.created_by() == dataset.created_by()
+    assert retrieved_dataset.organization() == dataset.organization()
 
     dataset = client.get_dataset(dataset.uid)
     assert dataset.name == name
@@ -48,9 +40,6 @@ def test_dataset(client, rand_gen):
     assert dataset.description == description
 
     dataset.delete()
-    final = list(client.get_datasets())
-    assert dataset not in final
-    assert set(final) == set(before)
 
     with pytest.raises(ResourceNotFoundError):
         dataset = client.get_dataset(dataset.uid)

tests/integration/test_pagination.py

@@ -0,0 +1,13 @@
+from copy import copy
+
+
+def test_get_one_and_many_dataset_order(client):
+    paginator = client.get_datasets()
+    # confirm get_one returns first dataset
+    all_datasets = list(paginator)
+    get_one_dataset = copy(paginator).get_one()
+    assert get_one_dataset.uid == all_datasets[0].uid
+
+    # confirm get_many(1) returns first dataset
+    get_many_datasets = copy(paginator).get_many(1)
+    assert get_many_datasets[0].uid == all_datasets[0].uid