refactor: add tests for ImagineLA and CAEDD

app/tests/src/test_chat_engine.py

@@ -1,4 +1,5 @@
 from src import chat_engine
+from src.chat_engine import CaEddWebEngine, ImagineLaEngine
 
 
 def test_available_engines():
@@ -7,3 +8,19 @@ def test_available_engines():
     assert len(engines) > 0
     assert "ca-edd-web" in engines
     assert "imagine-la" in engines
+
+
+def test_create_engine_CA_EDD():
+    engine_id = "ca-edd-web"
+    engine = chat_engine.create_engine(engine_id)
+    assert engine is not None
+    assert engine.name == CaEddWebEngine.name
+    assert engine.datasets == ["CA EDD"]
+
+
+def test_create_engine_Imagine_LA():
+    engine_id = "imagine-la"
+    engine = chat_engine.create_engine(engine_id)
+    assert engine is not None
+    assert engine.name == ImagineLaEngine.name
+    assert engine.datasets == ["CA EDD", "Imagine LA"]