chore: unit test different power models and get results

Signed-off-by: Huamin Chen <[email protected]>

Author: rootfs

tests/estimator_power_request_test.py

@@ -5,17 +5,17 @@
 from kepler_model.util.train_types import (
     CATEGORICAL_LABEL_TO_VOCAB,
     SYSTEM_FEATURES,
-    WORKLOAD_FEATURES,
+    BPF_FEATURES,
     ModelOutputType,
 )
 from tests.extractor_test import test_energy_source
 
-trainer_names = ["SGDRegressorTrainer"]
+trainer_names = ["SGDRegressorTrainer_0", "LogarithmicRegressionTrainer_0.json"]
 test_energy_sources = ["acpi", "rapl-sysfs"]
 
 
 def generate_request(
-    train_name, n=1, metrics=WORKLOAD_FEATURES, system_features=SYSTEM_FEATURES, output_type=ModelOutputType.DynPower.name, energy_source=test_energy_source
+    train_name, n=1, metrics=BPF_FEATURES, system_features=SYSTEM_FEATURES, output_type=ModelOutputType.AbsPower.name, energy_source=test_energy_source
 ):
     request_json = dict()
     if train_name is not None:
@@ -25,19 +25,20 @@ def generate_request(
     request_json["system_values"] = []
     for m in system_features:
         request_json["system_values"] += [CATEGORICAL_LABEL_TO_VOCAB[m][0]]
-    request_json["values"] = [[1.0] * len(metrics)] * n
+    request_json["values"] = [[1000.0] * len(metrics)] * n
     request_json["output_type"] = output_type
     request_json["source"] = energy_source
     return request_json
 
 
 def process(client, energy_source):
-    request_json = generate_request(trainer_names[0], 2, output_type="AbsPower", energy_source=energy_source)
-    res = client.make_request(request_json)
-    res_json = json.loads(res)
-    print(res_json)
-    assert res_json["msg"] == "", "response error: {}".format(res_json["msg"])
-    assert len(res_json["powers"]) > 0, "zero powers"
+    for trainer_name in trainer_names:
+        request_json = generate_request(trainer_name, 2, output_type="AbsPower", energy_source=energy_source)
+        res = client.make_request(request_json)
+        res_json = json.loads(res)
+        print(res_json)
+        assert res_json["msg"] == "", "response error: {}".format(res_json["msg"])
+        assert len(res_json["powers"]) > 0, "zero powers"
 
 
 class Client: