1 files changed, 7 insertions, 11 deletions
diff --git a/Omni/Llamacpp.py b/Omni/Llamacpp.py
index eac08fd..c7e9078 100755
--- a/Omni/Llamacpp.py
+++ b/Omni/Llamacpp.py
@@ -9,7 +9,10 @@ sure llama-cpp still works in case I need/want to switch at some point.
 
 # : out llamacpp-test
 # : run llama-cpp
-
+import logging
+import Omni.App as App
+import Omni.Log as Log
+import Omni.Test as Test
 import os
 import sys
 import unittest
@@ -23,18 +26,11 @@ class TestLlamaCpp(unittest.TestCase):
         self.assertIn("llama-cpp", os.environ.get("PATH", ""))
 
 
-def test() -> None:
-    """Run this module's test suite."""
-    suite = unittest.TestSuite()
-    suite.addTests(
-        unittest.defaultTestLoader.loadTestsFromTestCase(TestLlamaCpp),
-    )
-    unittest.TextTestRunner().run(suite)
-
-
 def main() -> None:
     """Entrypoint."""
     if sys.argv[1] == "test":
-        test()
+        logger = logging.getLogger(__name__)
+        Log.setup(logger)
+        Test.run(App.Area.Test, [TestLlamaCpp])
     else:
         sys.exit(0)