Simplify document counting in unit tests

DataBiosphere · Sep 20, 2023 · b3dfea2 · b3dfea2
1 parent 514a2c9
commit b3dfea2
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/test/indexer/test_indexer.py b/test/indexer/test_indexer.py
@@ -407,11 +407,7 @@ def _walkthrough(v):
         return old_file_uuid
 
     def _num_docs_by_index(self, hits) -> Mapping[tuple[str, bool], int]:
-        counter = Counter()
-        for hit in hits:
-            entity_type, aggregate = self._parse_index_name(hit)
-            counter[entity_type, aggregate] += 1
-        return counter
+        return Counter(map(self._parse_index_name, hits))
 
     def test_indexed_matrices(self):
         """