Exemplo n.º 1
0
def test__fetch_simulation_config_emptyDict_returnsDefaultSettings():
    sim_config = {}
    config = Config()

    config._fetch_simulation_config(sim_config)

    assert (config.timedelta == 0 and config.iterations == 0
            and config.initial_population == 0
            and config.start_date == date(2000, 1, 1)
            and config.end_date == date(2000, 1, 1)
            and config.validations == [])
Exemplo n.º 2
0
def test__fetch_simulation_config_validData_fetchesValidVariables():
    sim_config = {
        "timedelta": 1,
        "iterations": 1,
        "initial_population": 1,
        "learning": {
            "start_date": "2010-01-01",
            "end_date": "2020-01-01"
        },
        "testing": [{
            "start_date": "2010-01-01",
            "end_date": "2020-01-01"
        }],
    }
    config = Config()

    config._fetch_simulation_config(sim_config)

    assert (config.timedelta == 1 and config.iterations == 1
            and config.initial_population == 1
            and config.start_date == date(2010, 1, 1)
            and config.end_date == date(2020, 1, 1)
            and config.validations[0][0] == date(2010, 1, 1)
            and config.validations[0][1] == date(2020, 1, 1))