Skip to content

Commit

Permalink
fix(test): fix check data test
Browse files Browse the repository at this point in the history
  • Loading branch information
roedoejet committed Jan 22, 2025
1 parent 2e0a40b commit ba627a7
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions fs2/tests/test_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
TargetTrainingTextRepresentationLevel,
)
from everyvoice.tests.basic_test_case import BasicTestCase
from everyvoice.tests.preprocessed_audio_fixture import PreprocessedAudioFixture
from everyvoice.tests.stubs import capture_stderr, silence_c_stderr
from everyvoice.utils import generic_psv_filelist_reader
from typer.testing import CliRunner
Expand Down Expand Up @@ -296,12 +297,13 @@ def test_not_multispeaker_with_speaker(self):
)


class CLITest(TestCase):
class CLITest(PreprocessedAudioFixture, BasicTestCase):
"""
Validate that all subcommands are accessible.
"""

def setUp(self) -> None:
super().setUp()
self.runner = CliRunner()
self.subcommands = (
"benchmark",
Expand All @@ -314,7 +316,7 @@ def test_check_data(self):
filelist = generic_psv_filelist_reader(BasicTestCase.data_dir / "metadata.psv")
with capture_stderr():
checked_data = check_data_from_filelist(
filelist, heavy_objective_evaluation=True
self.preprocessor, filelist, heavy_objective_evaluation=True
)
self.assertIn("pesq", checked_data[0])
self.assertIn("stoi", checked_data[0])
Expand Down

0 comments on commit ba627a7

Please sign in to comment.