diff --git a/tests/estimator_power_request_test.py b/tests/estimator_power_request_test.py index dae2d781..ff0f904e 100644 --- a/tests/estimator_power_request_test.py +++ b/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: