Release 0.0.2

[ric-app/ad.git] / tests / test_ad.py

diff --git a/tests/test_ad.py b/tests/test_ad.py
index c60c9dc..70de7dd 100644 (file)
--- a/tests/test_ad.py
+++ b/tests/test_ad.py
@@ -21,13 +21,18 @@ from ad.ad_train import train
 import json
 
 
-def test_RFtrainmodel(monkeypatch):
-    if not os.path.isfile('ad/RF'):
-        train()
+def test_database_connection(monkeypatch):
+    # start qp
+    main.connectdb(thread=True)
 
 
-def test_predict_anomaly(monkeypatch):
-    main.predict_anomaly('test')
+def test_trainModel(monkeypatch):
+    if not os.path.isfile('model'):
+        train(thread=True)
+
+
+def test_predict_anomaly(monkeypatch, ad_ue):
+    main.predict_anomaly(monkeypatch, ad_ue)
 
 
 def test_msg_to_ts(monkeypatch, ad_to_ts):