Add another fix to models tests

marl · Nov 20, 2018 · 9a6885f · 9a6885f
1 parent 4933b09
commit 9a6885f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -3,22 +3,22 @@
 
 def test_get_embedding_model_path():
     embedding_model_path = get_embedding_model_path('linear', 'music')
-    assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_linear_music.h5'
+    assert '/'.join(embedding_model_path.split('/')[-2:]) == 'openl3/openl3_audio_linear_music.h5'
 
     embedding_model_path = get_embedding_model_path('linear', 'env')
-    assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_linear_env.h5'
+    assert '/'.join(embedding_model_path.split('/')[-2:]) == 'openl3/openl3_audio_linear_env.h5'
 
     embedding_model_path = get_embedding_model_path('mel128', 'music')
-    assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel128_music.h5'
+    assert '/'.join(embedding_model_path.split('/')[-2:]) == 'openl3/openl3_audio_mel128_music.h5'
 
     embedding_model_path = get_embedding_model_path('mel128', 'env')
-    assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel128_env.h5'
+    assert '/'.join(embedding_model_path.split('/')[-2:]) == 'openl3/openl3_audio_mel128_env.h5'
 
     embedding_model_path = get_embedding_model_path('mel256', 'music')
-    assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel256_music.h5'
+    assert '/'.join(embedding_model_path.split('/')[-2:]) == 'openl3/openl3_audio_mel256_music.h5'
 
     embedding_model_path = get_embedding_model_path('mel256', 'env')
-    assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel256_env.h5'
+    assert '/'.join(embedding_model_path.split('/')[-2:]) == 'openl3/openl3_audio_mel256_env.h5'
 
 
 def test_get_embedding_model():