def test_submission_model_dir(copy_model_dir):
tmp_model_dir = copy_model_dir
for fname in [
'model_fn.pkl', 'model_params.pkl', 'global_params.pkl',
'model_weights.h5.xz'
]:
assert os.path.isfile(os.path.join(tmp_model_dir, fname))
if os.path.isfile(os.path.join(tmp_model_dir, 'model_weights.h5')):
os.remove(os.path.join(tmp_model_dir, 'model_weights.h5'))
assert not does_model_exists(tmp_model_dir)
decompress_model_if_necessary(tmp_model_dir)
assert does_model_exists(tmp_model_dir)
def test_prepare_model_from_solution(copy_model_dir, code_dir, fake_input_dir):
configs = {
'code_dir': code_dir,
'input_dir': fake_input_dir,
'model_dir': copy_model_dir,
'HORIZON': 12,
}
prepare_model(configs)
assert does_model_exists(copy_model_dir)
def test_train_model(tmpdir, horizon, code_dir, fake_input_dir):
model_dir = str(tmpdir)
configs = {
'code_dir': code_dir,
'input_dir': fake_input_dir,
'model_dir': model_dir,
'HORIZON': horizon,
}
prepare_model(configs, test_run=True)
assert does_model_exists(model_dir)
model = EuropaModel.restore_model(model_dir)
assert model.horizon == horizon
def test_load_model(ready_model_dir):
assert does_model_exists(ready_model_dir)
model = EuropaModel.restore_model(ready_model_dir)
assert isinstance(model, EuropaModel)
def test_missing_element(fake_model_dir, missing_element):
os.remove(os.path.join(fake_model_dir, missing_element))
assert not does_model_exists(fake_model_dir)
def test_does_model_exists(fake_model_dir):
assert does_model_exists(fake_model_dir)