Merge pull request #35 from data-engineering-collective/issue-28

Ensure put returns key
data-engineering-collective · May 6, 2021 · de4804e · de4804e
2 parents 1ecec4f + bd0969a
commit de4804e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/basic_store.py b/tests/basic_store.py
@@ -20,8 +20,8 @@ def is_emulated_gcstore_test(store):
 
 class BasicStore(object):
     def test_store(self, store, key, value):
-        key = store.put(key, value)
-        assert isinstance(key, str)
+        new_key = store.put(key, value)
+        assert key == new_key
 
     def test_unicode_store(self, store, key, unicode_value):
         with pytest.raises(IOError):