From 2599e2a6a43930e116ab613226fca7d8758c25b8 Mon Sep 17 00:00:00 2001 From: Danilo Lessa Bernardineli Date: Thu, 18 Jan 2024 20:34:39 -0300 Subject: [PATCH] update tests --- testing/test_param_count.py | 12 ++++++++++-- testing/tests/cadCAD_memory_address.json | 2 +- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/testing/test_param_count.py b/testing/test_param_count.py index 64eedb83..317bcb04 100644 --- a/testing/test_param_count.py +++ b/testing/test_param_count.py @@ -25,15 +25,23 @@ def run_experiment(exp: Experiment, mode: str): return records -def param_count_test_generator(provided_params): +def param_count_test_suf_generator(provided_params): def s_test_param_count(params, _2, _3, _4, _5): assert params.keys() == provided_params.keys(), 'Params are not matching' + return ('varA', None) + return s_test_param_count +def param_count_test_policy_generator(provided_params): + def p_test_param_count(params, _2, _3, _4): + assert params.keys() == provided_params.keys(), 'Params are not matching' + return {'sigA': None} + return p_test_param_count + def create_experiments(N_simulations=3, N_sweeps=3, N_runs=3, N_timesteps=3, N_substeps=3, params={}) -> Experiment: INITIAL_STATE = {'varA': None} - PSUBs = [{'policies': {}, 'variables': {'varA': param_count_test_generator(params)}}] * N_substeps + PSUBs = [{'policies': {'sigA': param_count_test_policy_generator(params)}, 'variables': {'varA': param_count_test_suf_generator(params)}}] * N_substeps SIM_CONFIG = config_sim( { diff --git a/testing/tests/cadCAD_memory_address.json b/testing/tests/cadCAD_memory_address.json index d6405c8c..322a5c64 100644 --- a/testing/tests/cadCAD_memory_address.json +++ b/testing/tests/cadCAD_memory_address.json @@ -1 +1 @@ -{"memory_address": "0x10d841d50"} \ No newline at end of file +{"memory_address": "0x111444900"} \ No newline at end of file