tests

Kepler Model Server Tests

• Kepler Model Server Tests
  • Unit Tests
    • Promethes Query
    • Extractor
    • Isolator
    • Trainer
    • Pipeline
    • Estimator Power Prediction
    • Estimator Model Request (to Model Server)
    • Estimator Power Request (from Collector)
    • Offline Trainer
    • Server API
  • Integration Test

Unit Tests

Promethes Query

The test is for testing making query to prometheus server. The output will be saved in json format.

Requirements:

• kepler-exporter running and export metrics to prometheus server
• Prometheus server serves at localhost:9090

Run:

python prom_test.py

Reuse:

from prom_test import process as prom_process
prom_process()

Optional arguments:

• server: prometheus server address (default: localhost:9090)
• save_path: path to save a query response (default: data/prom_output)
• save_name: response filename to save (default: prom_response.json)
• interval: query interval (default: 300s)
• step: query step (default: 3s)

Extractor

The test is for testing extractors from saved prometheus response. The output will be saved in csv format for node-level and container-level data of each feature group.

Requirements:

• prometheus response in json (default: data/prom_output/prom_repsonse.json)

Run:

python extractor_test.py

Reuse:

from extractor_test import process as extractor_process
extractor_process(query_results, feature_group)

Optional arguments:

• save_path: path to save a extracted result (default: data/extractor_output)
• customize_extractors: additional extractor names (default: [])
• energy_source: target energy source (default: rapl-sysfs)
• num_of_unit: number of component units (default: 2)

Isolator

The test is for testing isolators from container-level extracted data. The output will be saved in csv format for container-level data of each feature group for each extractor.

Requirements:

• extractor output (default: data/extractor_output)

Run:

python isolator_test.py

Reuse:

from isolator_test import process as isolate_process
isolate_process()

Optional arguments:

• save_path: path to save a isolated result (default: data/isolator_output)
• extract_path: input path of extracted result (default: data/extractor_output)
• test_isolators: list of basic isolator instances (default: [MinIsolator, NoneIsolator])
• customize_isolators: additional isolator instances (default: [])

Trainer

The test is for testing trainers for node-level power from extracted data and for container-level power from isolated data. The trained models will be saved in model path (default: src/models) and accuracy results will be printed.

Requirements:

• extractor output (default: data/extractor_output)
• isolator output (default: data/isolator_output)

Run

python trainer_test.py

Reuse:

from isolator_test import get_isolate_results
from extractor_test import get_extract_results
from trainer_test import process as trainer_process

# node-level, use extractor_output at extractor_output_path (default: data/extractor_output)
extractor_results = get_extract_results(extractor_name, node_level=True, save_path=extractor_output_path)
for feature_group, result in extractor_results.items():
    trainer_process(True, feature_group, result)

# container-level, use isolator_output at isolator_output_path (default: data/isolator_output)
isolator_results = get_isolate_results(isolator_name, extractor_name, save_path=isolator_output_path)
for feature_group, result in isolator_results.items():
    trainer_process(False, feature_group, result)

Optional arguments:

• trainer_names: trainer names to be initiated (default: test_trainer_names)
• energy_source: target energy source (default: rapl-sysfs)
• power_columns: extracted power columns (default: generated power columns from prom_response)
• pipeline_name: pipeline name to be saved under model top path, /models, (default: default)

Pipeline

The test is for testing pipeline execution which is composed of testing extractor, testing isolator, and testing trainer. The input is saved prometheus response. The trained models will be saved in model path (default: src/models)/

Requirements:

• prometheus response in json (default: data/prom_output/prom_repsonse.json)

Run:

python pipeline_test.py

Reuse:

from pipeline_test import process as pipeline_process
pipeline_process()

Optional arguments:

• prom_save_path: path of prometheus response (default: data/prom_output)
• prom_save_name: prometheus response filename (default: prom_response)
• target_energy_sources: list of target energy sources (default: [rapl-sysfs])
• extractors: list of extractor instances (default: [DefaultExtractor])
• isolators: list of isolator instances (default: [MinIsolator, NoneIsolator])
• abs_trainer_names: list of trainer names to learn for DynPower (default: test_trainer_names)
• dyn_trainer_names: list of trainer names to learn for AbsPower (default: test_trainer_names)

Estimator Power Prediction

The test is for testing applying a trained model to extracted and isolated data for node-level and container-level power, respectively. Prediction error will be printed.

Requirements:

• trained models (default: ../src/models/default)

Run:

python estimator_model_test.py

Reuse (test_model):

from util import ModelOutputType, model_toppath, default_train_output_pipeline
from isolator_test import get_isolate_results
from extractor_test import get_extract_results, get_expected_power_columns

from estimator_model_test import test_model

# model folder under model_toppath which is ../src/models by default
pipeline_name = default_train_output_pipeline
# get_expected_power_columns(energy_components=test_energy_components, num_of_unit=test_num_of_unit)
power_columns = get_expected_power_columns()

# node-level, use extractor_output at extractor_output_path (default: data/extractor_output)
extractor_results = get_extract_results(extractor_name, node_level=True, save_path=extractor_output_path)
for feature_group, result in extractor_results.items():
   group_path = get_model_group_path(model_toppath, ModelOutputType.AbsPower, feature_group, energy_source, pipeline_name)
   test_model(group_path, model_name, result, power_columns)

# container-level, use isolator_output at isolator_output_path (default: data/isolator_output)
isolator_results = get_isolate_results(isolator_name, extractor_name, save_path=isolator_output_path)
for feature_group, result in isolator_results.items():
   group_path = get_model_group_path(model_toppath, ModelOutputType.DynPower, feature_group, energy_source, pipeline_name)
   test_model(group_path, model_name, result, power_columns)

Optional arguments:

• power_range: power range to compute scaled error in percentage

Estimator Model Request (to Model Server)

The test is for testing estimator as a client to model server by making a request for best model for a specific feature group and performing a prediction by the loaded model. Output of prediction will be printed.

Requirements:

• trained model
• server running

python ../src/server/model_server.py

Run:

python estimator_model_request_test.py

Estimator Power Request (from Collector)

The test is for testing estimator as a server to serve a power request from Kepler collector. Response output will be printed

Requirements:

• model initial url defined in util/loader.py is available.
• estimator running

python ../src/estimate/estimator.py

Run:

python estimator_power_request_test.py

Offline Trainer

The test is for testing offline trainer as a server to serve a training pipeline. Response output will be in data/offline_trainer_output.

Requirements:

• offline trainer server running

python ../src/train/offline_trainer.py

Run:

python offline_trainer_test.py

Reuse:

from offline_trainer_test import process as offline_trainer_client_process
dataset_name = # pipeline prefix such as dataset name
# prom_response can be obtained by reusing prom_test to query data from metric server
#   from prom_test import process as prom_process
#   prom_process()
# idle_prom_response is used for ProfileBackgroundIsolator, to remove background power, and TrainIsolator, to separate background container.
train_prom_response = # prom_response when target container for training is running 
idle_prom_response = # prom_response when no target container
offline_trainer_client_process(dataset_name, train_prom_response, idle_prom_response)

Optional arguments:

• energy_source: target energy source (default: rapl-sysfs)
• isolators: dict map of isolator class name to argument dict map (default: {"MinIdleIsolator": {}, "NoneIsolator": {}, "ProfileBackgroundIsolator": {}, "TrainIsolator": {"abs_pipeline_name": default_train_output_pipeline}})
• target_path: path to save trained ouput (default: data/offline_trainer_output)

Server API

The test is for testing server model selection API.

Requirements:

• server running

# create new model folder
mkdir -p $(pwd)/db-models
MODEL_PATH=$(pwd)/db-models model-server

Run:

hatch run test -vvv -s ./tests/model_select_test.py

Integration Test

./e2e_test.sh <options>

scenario	test cases	deploy options	status
Kepler with estimator only	Latest Kepler image is connected to estimator	--estimator	✔
	Estimator can load model from config	--estimator	✔
	Dummy power request can be made to estimator	--test --estimator	✔
Kepler with model server only	Latest Kepler image can request weight model from model server	--server	✔
	Dummy weight model request can be made to model server	--test --server	✔
Kepler with estimator and model server	Estimator can load model from model server	--estimator --server	✔
	Model server can load initial model	--estimator --server	✔
	Dummy power request can be made to estimator	--test --estimator --server	✔
Kepler with model server and online trainer	Kepler can provide metric to online trainer to train	ONLINE	WIP
	Dummy prometheus server can provide metric to online trainer to train	ONLINE TEST	WIP
	Trained model is updated to pipeline and availble on model server		WIP