820de79517aaed577f9af9131f5ec87cd432f04a,mltsp/tests/test_custom_feats.py,,teardown,#,132

Before Change

def teardown():
    Tear-down - remove tmp files
    if os.path.exists(tmp_dir_path):
        shutil.rmtree(tmp_dir_path, ignore_errors=True)
    assert(not os.path.exists(tmp_dir_path))

    for f in [os.path.join(cfg.MLTSP_PACKAGE_PATH,
                           "custom_feature_scripts/custom_feature_defs.py")]:

After Change

    cft.remove_tmp_files_and_container("test", "/tmp/mltsp_test")
    assert(not os.path.exists("/tmp/mltsp_test"))
    for tmp_file in [os.path.join(cfg.TMP_CUSTOM_FEATS_FOLDER,
                                  "custom_feature_defs.py"),
                     os.path.join(cfg.TMP_CUSTOM_FEATS_FOLDER,
                                  "custom_feature_defs.pyc"),
                     os.path.join(cfg.TMP_CUSTOM_FEATS_FOLDER,
                                  "__init__.pyc"),
                     os.path.join(os.path.join(cfg.PROJECT_PATH,

Instances