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)