ati: ATI tests split by activity/organisation

IATI · Dec 13, 2024 · 7239ad7 · 7239ad7
1 parent cf1d4f0
commit 7239ad7
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/stats/analytics.py b/stats/analytics.py
@@ -529,14 +529,16 @@ def _ruleset_passes(self):
         return out
 
 
-    @returns_numberdict
+    @returns_numberdictdict
     @memoize
     def ati_tests(self):
-        out = {}
+        out = defaultdict(dict)
+        tag = self.element.tag
         for test in ati_tests:
-            result = test(self.element, codelists=CODELISTS[self._major_version()])
-            result = int(bool(result))
-            out[f"{test.feature.name}: {test.name}"] = result
+            if tag in test.feature.tags:
+                result = test(self.element, codelists=CODELISTS[self._major_version()])
+                result = int(bool(result))
+                out[tag][f"{test.feature.name}: {test.name}"] = result
         return out
 
     @returns_number