tests/test_adapter_switching.py

import sys
import os
import unittest
from pathlib import Path

# Add project root to path
sys.path.insert(0, os.getcwd())

from ankahi_backend.model_loader import ModelLoader

class TestAdapterSwitching(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.loader = ModelLoader()
        cls.loader.load_base_model()

    def test_switch_between_personas(self):
        # Switch to Arjun
        self.loader.load_persona_adapter("arjun")
        self.assertEqual(self.loader.active_persona, "arjun")
        
        # Check some weights change (simple heuristic)
        # In a real test we'd check specific layer weights
        
        # Switch to Ananya
        self.loader.load_persona_adapter("ananya")
        self.assertEqual(self.loader.active_persona, "ananya")
        
        # Unload
        self.loader.unload_adapter()
        self.assertIsNone(self.loader.active_persona)

if __name__ == "__main__":
    unittest.main()