From 26c1e380970dfec67b053a865539767fadc2775e Mon Sep 17 00:00:00 2001
From: OpenHTF Owners <no-reply@google.com>
Date: Mon, 26 Feb 2024 11:40:00 -0800
Subject: [PATCH] Added pytype None checks to test.py.

PiperOrigin-RevId: 610475779
---
 openhtf/util/test.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/openhtf/util/test.py b/openhtf/util/test.py
index 4f8825800..54cabb7b8 100644
--- 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