diff tests/test_load_features.py @ 43:b1d9ba5f888e branch-tests

debugging tests

--- a/tests/test_load_features.py	Fri Sep 15 12:27:11 2017 +0100
+++ b/tests/test_load_features.py	Fri Sep 15 16:17:55 2017 +0100
@@ -14,6 +14,13 @@
 
 feat_loader = load_features.FeatureLoader(win2sec=8)
 
+#TEST_METADATA_FILE = '../data/sample_dataset/metadata.csv'
+TEST_METADATA_FILE = os.path.join(os.path.dirname(__file__), os.path.pardir, 
+                                    'data', 'sample_dataset', 'metadata.csv')
+#TEST_METADATA_FILE = 'data/sample_dataset/metadata.csv'
+#TEST_MELODIA_FILE = 'data/sample_dataset/Melodia/mel_1_2_1.csv'
+TEST_MELODIA_FILE = os.path.join(os.path.dirname(__file__), os.path.pardir, 
+                                 'data', 'sample_dataset', 'Melodia', 'mel_1_2_1.csv')
 
 def test_get_music_idx_from_bounds():
     bounds = np.array([['0', '10.5', 'm']])
@@ -213,45 +220,49 @@
     
 
 def test_get_pb_for_file_n_bins():
-    pbihist = feat_loader.get_pb_for_file('data/sample_dataset/Melodia/mel_1_2_1.csv', nmfpb=False, scale=False)
+    pbihist = feat_loader.get_pb_for_file(TEST_MELODIA_FILE, nmfpb=False, scale=False)
     assert pbihist.shape[1] == 3600
 
 
 def test_get_pb_for_file_align():
-    pbihist = feat_loader.get_pb_for_file('data/sample_dataset/Melodia/mel_1_2_1.csv', nmfpb=False, scale=False)
+    pbihist = feat_loader.get_pb_for_file(TEST_MELODIA_FILE, nmfpb=False, scale=False)
     pbihist = pbihist.get_values()
     assert np.sum(pbihist[:, :60].ravel()) > np.sum(pbihist[:, 60:120].ravel())
 
 
 def test_get_pb_for_file_nmf():
-    pbihist = feat_loader.get_pb_for_file('data/sample_dataset/Melodia/mel_1_2_1.csv', nmfpb=True, scale=False)
+    pbihist = feat_loader.get_pb_for_file(TEST_MELODIA_FILE, nmfpb=True, scale=False)
     assert pbihist.shape[1] == 240
 
 
 def test_get_features():
-    df = pd.read_csv('data/sample_dataset/metadata.csv')
+    df = pd.read_csv(TEST_METADATA_FILE)
     df = df.iloc[:1, :]
     os.chdir('data/')
-    data_list = feat_loader.get_features(df)
+    print df.head()
+    print os.getcwd()
+    ddf = pd.read_csv(df['Melodia'].iloc[0])
+    print ddf.head()
+    data_list = feat_loader.get_features(df, precomp_melody=False)
     os.chdir('..')
     assert len(np.unique(data_list[-1])) == 1
 
 
 def test_get_features_n_files():
-    df = pd.read_csv('data/sample_dataset/metadata.csv')
+    df = pd.read_csv(TEST_METADATA_FILE)
     n_files = 3
     df = df.iloc[:n_files, :]
     os.chdir('data/')
-    data_list = feat_loader.get_features(df)
+    data_list = feat_loader.get_features(df, precomp_melody=False)
     os.chdir('..')
     assert len(np.unique(data_list[-1])) == n_files
 
 
 def test_get_features_n_frames():
-    df = pd.read_csv('data/sample_dataset/metadata.csv')
+    df = pd.read_csv(TEST_METADATA_FILE)
     df = df.iloc[:1, :]
     os.chdir('data/')
-    data_list = feat_loader.get_features(df)
+    data_list = feat_loader.get_features(df, precomp_melody=False)
     os.chdir('..')
     dur_sec = 11.5  # duration of first file in metadata.csv is > 11 seconds
     n_frames_true = np.round((dur_sec - feat_loader.win2sec) * feat_loader.framessr2)