Extra tests for Travis

Author: chriswang030

.travis.yml

@@ -19,6 +19,7 @@ install:
 # run tests
 script:
   - export PYTHONPATH=$PYTHONPATH:./
+  - curl 'https://zenodo.org/record/3345256/files/mad.pt?download=1' > autosynch/mad_twinnet/outputs/states/mad.pt
   - pytest --cov-config=.coveragerc --cov=autosynch tests/
 
 # only test phase2

README.md

@@ -1,5 +1,7 @@
 # autosynch repository
 
+[![Build Status](https://travis-ci.com/chriswang030/autosynch.svg?branch=phase2)](https://travis-ci.com/chriswang030/autosynch)
+
 WORK IN PROGRESS
 
 This branch will contain additional work done in phase 2 of GSoC, including:
@@ -8,4 +10,5 @@ This branch will contain additional work done in phase 2 of GSoC, including:
 - syllable alignment of music and lyrics
 - initial testing for real-time processing
 
-To get weights trained on MedleyDB_V1, download it here: [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.3334973.svg)](https://doi.org/10.5281/zenodo.3334973)
+Get weights trained on MedleyDB_V1 here: [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.3334973.svg)](https://doi.org/10.5281/zenodo.3334973)
+Get weights trained on MedleyDB_V2 here: [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.3345256.svg)](https://doi.org/10.5281/zenodo.3345256)

tests/test_twinnet.py

@@ -21,11 +21,10 @@ def test_generate_output_path(audio_file):
     assert twinnet._make_target_file_names(['1&*!!.wav']) == [['1&*!!_voice.wav', '1&*!!_bg_music.wav']]
     assert twinnet._make_target_file_names(['1', '2']) == [['1_voice.wav', '1_bg_music.wav'], ['2_voice.wav', '2_bg_music.wav']]
 
-# Commented out for Travis since `mad.pt` not in Github repo
-# def test_twinnet_process_writes_to_file(audio_file, tmpdir):
-#     """Requires `mad.pt` in `mad_twinnet/outputs/states` to work"""
-#     out_base = os.path.splitext(os.path.basename(audio_file))[0] + '_voice.wav'
-#     out_path = os.path.join(str(tmpdir), out_base)
-#
-#     assert twinnet.twinnet_process(audio_file, [[out_path, '']]) is True
-#     assert os.path.isfile(out_path)
+def test_twinnet_process_writes_to_file(audio_file, tmpdir):
+    """Requires `mad.pt` in `mad_twinnet/outputs/states` to work"""
+    out_base = os.path.splitext(os.path.basename(audio_file))[0] + '_voice.wav'
+    out_path = os.path.join(str(tmpdir), out_base)
+
+    assert twinnet.twinnet_process(audio_file, [[out_path, '']]) is True
+    assert os.path.isfile(out_path)