Browse Source

[to #42322933] test: unify kws pipeline input type to AUDIO

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10362437
master
bin.xue yingda.chen 3 years ago
parent
commit
0d97f8959d
1 changed files with 2 additions and 17 deletions
  1. +2
    -17
      tests/pipelines/test_key_word_spotting_farfield.py

+ 2
- 17
tests/pipelines/test_key_word_spotting_farfield.py View File

@@ -22,18 +22,14 @@ class KWSFarfieldTest(unittest.TestCase):
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
def test_normal(self):
kws = pipeline(Tasks.keyword_spotting, model=self.model_id)
inputs = {'input_file': os.path.join(os.getcwd(), TEST_SPEECH_FILE)}
result = kws(inputs)
result = kws(os.path.join(os.getcwd(), TEST_SPEECH_FILE))
self.assertEqual(len(result['kws_list']), 5)
print(result['kws_list'][-1])

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
def test_mono(self):
kws = pipeline(Tasks.keyword_spotting, model=self.model_id)
inputs = {
'input_file': os.path.join(os.getcwd(), TEST_SPEECH_FILE_MONO)
}
result = kws(inputs)
result = kws(os.path.join(os.getcwd(), TEST_SPEECH_FILE_MONO))
self.assertEqual(len(result['kws_list']), 5)
print(result['kws_list'][-1])

@@ -44,17 +40,6 @@ class KWSFarfieldTest(unittest.TestCase):
self.assertEqual(len(result['kws_list']), 5)
print(result['kws_list'][-1])

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
def test_output(self):
kws = pipeline(Tasks.keyword_spotting, model=self.model_id)
inputs = {
'input_file': os.path.join(os.getcwd(), TEST_SPEECH_FILE),
'output_file': 'output.wav'
}
result = kws(inputs)
self.assertEqual(len(result['kws_list']), 5)
print(result['kws_list'][-1])

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
def test_input_bytes(self):
with open(os.path.join(os.getcwd(), TEST_SPEECH_FILE), 'rb') as f:


Loading…
Cancel
Save