From 9a6885f5d9a0886344b5c6b55cdcbfaee38a95e7 Mon Sep 17 00:00:00 2001 From: Jason Cramer Date: Mon, 19 Nov 2018 23:36:15 -0500 Subject: [PATCH] Add another fix to models tests --- tests/test_models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index d2afd09..0273086 100644 --- 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():