Skip to content

Commit f955a4b

Browse files
committed
add GridFS count fix by @Zhuoying, restore materialsvirtuallab pmg docker image
1 parent 9cdda25 commit f955a4b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

atomate/vasp/firetasks/tests/test_lobster_tasks.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -157,12 +157,12 @@ def test_mongodb_more_files(self):
157157
self.assertListEqual(load_dict["output"]["chargespilling"], [0.008, 0.008])
158158
db = self.get_task_database()
159159
gfs = gridfs.GridFS(db, "lobster_files")
160-
results = gfs.find({}).count()
161-
self.assertEqual(results, 2)
160+
results = gfs.find({})
161+
self.assertEqual(len(list(results)), 2)
162162
for fn in ["ICOHPLIST", "COOPCAR"]:
163163
oid = load_dict[fn.lower() + "_id"]
164-
results = gfs.find({"_id": oid}).count()
165-
self.assertEqual(results, 1)
164+
results = gfs.find({"_id": oid})
165+
self.assertEqual(len(list(results)), 1)
166166

167167
def test_jsonfile_si(self):
168168
copy_r(self.vasp_si_dir, ".")

0 commit comments

Comments
 (0)