Fix mock opensearch tests

libAtoms · Jun 12, 2024 · 3826c10 · 3826c10
1 parent 1a386b6
commit 3826c10
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/tests/test_opensearch_mock.py b/tests/test_opensearch_mock.py
@@ -78,10 +78,9 @@ def test_push(self, abcd):
         assert isinstance(atoms_2, Atoms)
         atoms_2.set_cell([1, 1, 1])
 
+        abcd.refresh()
         result = AtomsModel(
-            None,
-            None,
-            abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
+            dict=abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
         ).to_ase()
         assert atoms_1 == result
         assert atoms_2 != result
@@ -120,14 +119,10 @@ def test_bulk(self, abcd):
         assert abcd.count() == 2
 
         result_1 = AtomsModel(
-            None,
-            None,
-            abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
+            dict=abcd.client.search(index="test_index")["hits"]["hits"][0]["_source"],
         ).to_ase()
         result_2 = AtomsModel(
-            None,
-            None,
-            abcd.client.search(index="test_index")["hits"]["hits"][1]["_source"],
+            dict=abcd.client.search(index="test_index")["hits"]["hits"][1]["_source"],
         ).to_ase()
         assert atoms_1 == result_1
         assert atoms_2 == result_2