From 4933b0939fb5b2d4c5d56e3efc962a4e4dc7fcfa Mon Sep 17 00:00:00 2001 From: Jason Cramer Date: Mon, 19 Nov 2018 23:20:10 -0500 Subject: [PATCH] Update model paths in 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 5f9f23d..d2afd09 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/models/openl3_audio_linear_music.h5' + assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_linear_music.h5' embedding_model_path = get_embedding_model_path('linear', 'env') - assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/models/openl3_audio_linear_env.h5' + assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_linear_env.h5' embedding_model_path = get_embedding_model_path('mel128', 'music') - assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/models/openl3_audio_mel128_music.h5' + assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel128_music.h5' embedding_model_path = get_embedding_model_path('mel128', 'env') - assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/models/openl3_audio_mel128_env.h5' + assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel128_env.h5' embedding_model_path = get_embedding_model_path('mel256', 'music') - assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/models/openl3_audio_mel256_music.h5' + assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel256_music.h5' embedding_model_path = get_embedding_model_path('mel256', 'env') - assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/models/openl3_audio_mel256_env.h5' + assert '/'.join(embedding_model_path.split('/')[-3:]) == 'openl3/openl3_audio_mel256_env.h5' def test_get_embedding_model():