Update file storage test to only store data once.

janpisl · Jun 27, 2018 · 19241b4 · 19241b4
1 parent 5774cd4
commit 19241b4
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/test_storage.py b/tests/test_storage.py
@@ -95,10 +95,11 @@ def test_store(self):
                              supported_formats=[FORMATS.GML])
         vector_output.file = get_vector_file()
 
-        self.assertEqual(len(self.storage.store(vector_output)), 3) 
-        self.assertEqual(self.storage.store(vector_output)[0], STORE_TYPE.PATH)
-        self.assertIsInstance(self.storage.store(vector_output)[1], str)
-        self.assertIsInstance(self.storage.store(vector_output)[2], str)
+        store_file = self.storage.store(vector_output)        
+        self.assertEqual(len(store_file), 3) 
+        self.assertEqual(store_file[0], STORE_TYPE.PATH)
+        self.assertIsInstance(store_file[1], str)
+        self.assertIsInstance(store_file[2], str)
 
 
 class PgStorageTestCase(unittest.TestCase):