Add test for EVP. Set training matrix to identity for equal results o…

…n each execution with dummy data.

Signed-off-by: Felix Zailskas <[email protected]>

src/evp/evp.py

@@ -14,7 +14,7 @@ def __init__(self) -> None:
         self.life_time_value_predictor = LinearRegression()
 
         all_leads = get_database().get_all_leads()
-        X = np.random.random((len(all_leads), len(all_leads)))
+        X = np.identity(len(all_leads))
         y_probability = np.array(
             [lead.lead_value.customer_probability for lead in all_leads]
         )

tests/test_evp.py

@@ -0,0 +1,19 @@
+# SPDX-License-Identifier: MIT
+# SPDX-FileCopyrightText: 2023 Felix Zailskas <[email protected]>
+
+import os
+import sys
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
+
+from database import get_database
+from database.models import LeadValue
+from evp.evp import EstimatedValuePredictor
+
+
+def test_estimate_value():
+    leads = get_database().get_all_leads()
+    evp = EstimatedValuePredictor()
+    for lead in leads:
+        value = evp.estimate_value(lead.lead_id)
+        assert type(value) == LeadValue