fix test

Canner · Jul 1, 2024 · bd2ae99 · bd2ae99
1 parent a051717
commit bd2ae99
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/wren-ai-service/tests/pytest/providers/test_loader.py b/wren-ai-service/tests/pytest/providers/test_loader.py
@@ -3,7 +3,7 @@
 
 def test_import_mods():
     loader.import_mods("src.providers")
-    assert len(loader.PROVIDERS) == 6
+    assert len(loader.PROVIDERS) == 7
 
 
 def test_get_provider():
@@ -29,3 +29,6 @@ def test_get_provider():
 
     provider = loader.get_provider("wren-ibis")
     assert provider.__name__ == "WrenIbis"
+
+    provider = loader.get_provider("wren-engine")
+    assert provider.__name__ == "WrenEngine"