use featurizer

Author: RobRossmiller-TG

tests/data/baseline/ml/fastRP.json.gz

@@ -1,6 +1,5 @@
 clear
-python3 test/setup.py &&
+# python3 test/setup.py &&
   python3 test/baseline/create_baselines.py &&
-  pytest test/test_centrality.py &&
+  # pytest test/test_centrality.py test/test_ml.py
   pytest test/test_ml.py
-# pytest

tests/run.sh

@@ -2,6 +2,5 @@
 import fast_rp_baseline
 
 if __name__ == "__main__":
-    # degree_cent_baseline.run()
-
+    degree_cent_baseline.run()
     fast_rp_baseline.run()

tests/test/baseline/create_baselines.py

@@ -1,26 +1,17 @@
 import gzip
 import json
-import os
 
 import numpy as np
 import pytest
 from dotenv import load_dotenv
-from pyTigerGraph import TigerGraphConnection
+
+from util import get_featurizer
 
 load_dotenv()
 data_path_root = "data"
 baseline_path_root = f"{data_path_root}/baseline"
 
 
-conn = TigerGraphConnection(
-    host=os.environ["HOST_NAME"],
-    graphname="Cora",
-    username=os.environ["USER_NAME"],
-    password=os.environ["PASS"],
-)
-conn.getToken()
-
-
 def cos_sim(x, y):
     x_mag = np.linalg.norm(x)
     y_mag = np.linalg.norm(y)
@@ -29,6 +20,8 @@ def cos_sim(x, y):
 
 
 class TestML:
+    feat = get_featurizer("Cora")
+
     def test_fastRP(self):
         params = {
             "v_type_set": ["Paper"],
@@ -47,10 +40,9 @@ def test_fastRP(self):
         with gzip.open(f"{baseline_path_root}/ml/fastRP.json.gz", "rb") as f:
             baseline = json.load(f)
 
-        result = conn.runInstalledQuery(
+        result = self.feat.runAlgorithm(
             "tg_fastRP",
             params=params,
-            usePost=True,
         )
         result = {
             v["v_id"]: v["attributes"]["res.@final_embedding_list"]
@@ -61,4 +53,4 @@ def test_fastRP(self):
             v = result[bk]
             sim = abs(cos_sim(v, bv))
             if sim < threshold:
-                pytest.fail(f"cos-sim of {bk} is {sim} (< threshold of {threshold}")
+                pytest.fail(f"cos-sim of ID: {bk} is {sim} (< threshold of {threshold}")