keep forgettng they removed my bae, pytest

openpilot/sunnypilot/modeld_v2/tests/test_models.py

@@ -1,23 +1,24 @@
 import os
+import unittest
 from unittest.mock import patch
-import pytest
-from tinygrad.device import Device
 from openpilot.common.file_chunker import open_file_chunked
 from openpilot.sunnypilot.modeld_v2.helpers import load_oob
+from tinygrad.device import Device
 
 
-class TestLegacyModels:
+class TestLegacyModels(unittest.TestCase):
   def test_legacy_model_load(self):
     base_name = os.environ.get("MODEL_BASE_NAME")
     if not base_name:
-      pytest.skip("MODEL_BASE_NAME env var not set, skipping integration test.")
+      raise unittest.SkipTest("MODEL_BASE_NAME env var not set, skipping integration test.")
     chunk_dir = os.environ.get("MODEL_CHUNK_DIR", "/tmp/model_chunks")
     base_path = os.path.join(chunk_dir, base_name)
 
     try:
       f = open_file_chunked(base_path)
-    except Exception as e:
-      pytest.fail(f"Failed to open chunked file {base_path}: {e}")
+    except Exception as error:
+      self.fail(f"Failed to open chunked file {base_path}: {error}")
+    self.addCleanup(f.close)
 
     real_getitem = Device.__class__.__getitem__
 
@@ -30,4 +31,4 @@ class TestLegacyModels:
       obj = load_oob(f)
 
     assert isinstance(obj, dict), "Parsed object is not a dictionary"
-    assert 'metadata' in obj, "Metadata key is missing"
+    assert "metadata" in obj, "Metadata key is missing"