Fix test breakage likely caused by an update to pytest

google · Jun 29, 2024 · 1b777e9 · 1b777e9
1 parent 865512a
commit 1b777e9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/openhtf/util/test.py b/openhtf/util/test.py
@@ -660,10 +660,11 @@ class TestCase(unittest.TestCase):
 
   def __init__(self, methodName=None):
     super(TestCase, self).__init__(methodName=methodName)
-    test_method = getattr(self, methodName)
-    if inspect.isgeneratorfunction(test_method):
-      raise ValueError('%s yields without @openhtf.util.test.yields_phases' %
-                       methodName)
+    if methodName != 'runTest':
+      test_method = getattr(self, methodName)
+      if inspect.isgeneratorfunction(test_method):
+        raise ValueError('%s yields without @openhtf.util.test.yields_phases' %
+                         methodName)
 
   def setUp(self):
     super(TestCase, self).setUp()