Simple typing changes, Ruff import formatter. p3

1 files changed, 21 insertions, 13 deletions

diff --git a/tests/grliq/models/test_forensic_data.py b/tests/grliq/models/test_forensic_data.py
index 653f9a9..4fbf962 100644
--- a/tests/grliq/models/test_forensic_data.py
+++ b/tests/grliq/models/test_forensic_data.py
@@ -1,42 +1,50 @@
+from typing import TYPE_CHECKING
+
 import pytest
 from pydantic import ValidationError
 
-from generalresearch.grliq.models.forensic_data import GrlIqData, Platform
+if TYPE_CHECKING:
+    from generalresearch.grliq.models.forensic_data import GrlIqData
 
 
 class TestGrlIqData:
 
-    def test_supported_fonts(self, grliq_data):
+    def test_supported_fonts(self, grliq_data: "GrlIqData"):
         s = grliq_data.supported_fonts_binary
         assert len(s) == 1043
         assert "Ubuntu" in grliq_data.supported_fonts
 
-    def test_battery(self, grliq_data):
+    def test_battery(self, grliq_data: "GrlIqData"):
         assert not grliq_data.battery_charging
         assert grliq_data.battery_level == 0.41
 
-    def test_base(self, grliq_data):
-        g: GrlIqData = grliq_data
-        assert g.timezone == "America/Los_Angeles"
-        assert g.platform == Platform.LINUX_X86_64
-        assert g.webgl_extensions
+    def test_base(self, grliq_data: "GrlIqData"):
+        from generalresearch.grliq.models.forensic_data import Platform
+
+        assert grliq_data.timezone == "America/Los_Angeles"
+        assert grliq_data.platform == Platform.LINUX_X86_64
+        assert grliq_data.webgl_extensions
         # ... more
 
-        assert g.results is None
-        assert g.category_result is None
+        assert grliq_data.results is None
+        assert grliq_data.category_result is None
+
+        s = grliq_data.model_dump_json()
+        from generalresearch.grliq.models.forensic_data import GrlIqData, Platform
 
-        s = g.model_dump_json()
         g2: GrlIqData = GrlIqData.model_validate_json(s)
 
         assert g2.results is None
         assert g2.category_result is None
 
-        assert g == g2
+        assert grliq_data == g2
 
     # Testing things that will cause a validation error, should only be
     # because something is "corrupt", not b/c the user is a baddie
-    def test_corrupt(self, grliq_data):
+    def test_corrupt(self, grliq_data: "GrlIqData"):
         """Test for timestamp and timezone offset mismatch validation."""
+        from generalresearch.grliq.models.forensic_data import GrlIqData
+
         d = grliq_data.model_dump(mode="json")
         d.update(
             {