From 67436ab5551ec39c008d1bdeba5d04291aaccb1c Mon Sep 17 00:00:00 2001 From: discountchubbs Date: Tue, 8 Sep 2026 17:08:28 -0700 Subject: [PATCH] keep forgettng they removed my bae, pytest --- .../sunnypilot/modeld_v2/tests/test_models.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/openpilot/sunnypilot/modeld_v2/tests/test_models.py b/openpilot/sunnypilot/modeld_v2/tests/test_models.py index 1109299c2f..c31c4c818a 100644 --- a/openpilot/sunnypilot/modeld_v2/tests/test_models.py +++ b/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"