From 1c87c25f974da97f2620bf9747bd557e4112240d Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Fri, 22 May 2026 15:24:18 +0000
Subject: [PATCH] Add test for is_rknn_compatible

Added a new file `frigate/test/test_rknn_converter.py` with test cases to comprehensively test the `is_rknn_compatible` method in `frigate/util/rknn_converter.py`. The tests cover standard flows like `get_soc_type` returning supported and unsupported SoCs, as well as testing valid vs invalid explicit and implicit model types. Dependencies such as `cv2`, `numpy`, and `peewee` are correctly mocked using `sys.modules` to make the test environment-agnostic.

Co-authored-by: manupawickramasinghe <73810867+manupawickramasinghe@users.noreply.github.com>
---
 frigate/test/test_rknn_converter.py | 54 +++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 frigate/test/test_rknn_converter.py

diff --git a/frigate/test/test_rknn_converter.py b/frigate/test/test_rknn_converter.py
new file mode 100644
index 0000000000..0ca2063578
--- /dev/null
+++ b/frigate/test/test_rknn_converter.py
@@ -0,0 +1,54 @@
+import sys
+import unittest
+from unittest.mock import MagicMock, patch
+
+# Mock dependencies before any imports
+sys.modules['cv2'] = MagicMock()
+sys.modules['numpy'] = MagicMock()
+sys.modules['peewee'] = MagicMock()
+sys.modules['playhouse'] = MagicMock()
+sys.modules['playhouse.sqlite_ext'] = MagicMock()
+sys.modules['playhouse.shortcuts'] = MagicMock()
+sys.modules['unidecode'] = MagicMock()
+sys.modules['PIL'] = MagicMock()
+sys.modules['tzlocal'] = MagicMock()
+
+from frigate.util.rknn_converter import is_rknn_compatible
+
+class TestIsRknnCompatible(unittest.TestCase):
+    @patch("frigate.util.rknn_converter.get_soc_type")
+    def test_no_soc(self, mock_get_soc_type):
+        mock_get_soc_type.return_value = None
+        self.assertFalse(is_rknn_compatible("test.onnx"))
+
+    @patch("frigate.util.rknn_converter.get_soc_type")
+    def test_unsupported_soc(self, mock_get_soc_type):
+        mock_get_soc_type.return_value = "bcm2711"
+        self.assertFalse(is_rknn_compatible("test.onnx"))
+
+    @patch("frigate.util.rknn_converter.get_soc_type")
+    def test_supported_soc_valid_model_type(self, mock_get_soc_type):
+        mock_get_soc_type.return_value = "rk3588"
+        self.assertTrue(is_rknn_compatible("test.onnx", model_type="yolox"))
+
+    @patch("frigate.util.rknn_converter.get_soc_type")
+    def test_supported_soc_invalid_model_type(self, mock_get_soc_type):
+        mock_get_soc_type.return_value = "rk3588"
+        self.assertFalse(is_rknn_compatible("test.onnx", model_type="invalid"))
+
+    @patch("frigate.util.rknn_converter.get_soc_type")
+    @patch("frigate.util.rknn_converter.get_rknn_model_type")
+    def test_supported_soc_valid_inferred_model_type(self, mock_get_rknn_model_type, mock_get_soc_type):
+        mock_get_soc_type.return_value = "rk3588"
+        mock_get_rknn_model_type.return_value = "yolox"
+        self.assertTrue(is_rknn_compatible("yolox.onnx"))
+
+    @patch("frigate.util.rknn_converter.get_soc_type")
+    @patch("frigate.util.rknn_converter.get_rknn_model_type")
+    def test_supported_soc_invalid_inferred_model_type(self, mock_get_rknn_model_type, mock_get_soc_type):
+        mock_get_soc_type.return_value = "rk3588"
+        mock_get_rknn_model_type.return_value = None
+        self.assertFalse(is_rknn_compatible("unknown.onnx"))
+
+if __name__ == '__main__':
+    unittest.main()