upadte DeprecationsReport to handle SubTests

swryan · May 17, 2024 · 0da8e86 · 0da8e86
1 parent 16c6d10
commit 0da8e86
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/testflo/deprecations.py b/testflo/deprecations.py
@@ -16,10 +16,11 @@ def get_iter(self, input_iter):
 
         deprecations = {}
 
-        for test in input_iter:
-            for msg, locs in test.deprecations.items():
-                deprecations[msg] = deprecations.get(msg, set()) | locs
-            yield test
+        for tests in input_iter:
+            for test in tests:
+                for msg, locs in test.deprecations.items():
+                    deprecations[msg] = deprecations.get(msg, set()) | locs
+                yield test
 
         report = self.generate_report(deprecations)