def test_dame_utils_files_one_level_drop_pwd_method_returns_correct_result( self): du = DameUtils() cwd = os.getcwd() self.assertEqual( sorted(du.files_one_level_drop_pwd(cwd + "/files/datamodels")), [ 'files/datamodels/bernoulliNB_model.sav', 'files/datamodels/gaussianNB_model.sav', 'files/datamodels/multinomialNB_model.sav', 'files/datamodels/sgd_model.sav', 'files/datamodels/svc_model.sav' ])
def test_dame_utils_files_one_level_drop_pwd(self): du = DameUtils() cwd = os.getcwd() self.assertEqual( sorted( du.files_one_level_drop_pwd(cwd + "/files/datamodels/*sav")), [ 'files/datamodels/adaboost_model.sav', 'files/datamodels/bernoulliNB_model.sav', 'files/datamodels/forest_model.sav', 'files/datamodels/gaussianNB_model.sav', 'files/datamodels/mlp_model.sav', 'files/datamodels/multinomialNB_model.sav', 'files/datamodels/nltk_model.sav', 'files/datamodels/sgd_model.sav', 'files/datamodels/svc_model.sav', 'files/datamodels/tree_model.sav' ])