revert GridFS.count_documents({}) (doesn't exist) to GridFS.find({}).count()

Author: janosh

atomate/vasp/firetasks/tests/test_lobster_tasks.py

@@ -157,11 +157,11 @@ def test_mongodb_more_files(self):
         self.assertListEqual(load_dict["output"]["chargespilling"], [0.008, 0.008])
         db = self.get_task_database()
         gfs = gridfs.GridFS(db, "lobster_files")
-        results = gfs.count_documents({})
+        results = gfs.find({}).count()
         self.assertEqual(results, 2)
         for fn in ["ICOHPLIST", "COOPCAR"]:
             oid = load_dict[fn.lower() + "_id"]
-            results = gfs.count_documents({"_id": oid})
+            results = gfs.find({"_id": oid}).count()
             self.assertEqual(results, 1)
 
     def test_jsonfile_si(self):