Added pytype None checks to test.py.

PiperOrigin-RevId: 610475779
google · Feb 26, 2024 · 26c1e38 · 26c1e38
1 parent d1ebe96
commit 26c1e38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/openhtf/util/test.py b/openhtf/util/test.py
@@ -954,6 +954,7 @@ def assertAttachment(self,
 
   def get_diagnoses_store(self):
     self.assertIsNotNone(self.last_test_state)
+    assert self.last_test_state is not None
     return self.last_test_state.diagnoses_manager.store
 
   @classmethod