From d20b7a166fdfcaee6ab0fcbd6c58c5d66401f8a4 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 5 Oct 2022 20:35:17 +0530 Subject: [PATCH] tests --- tests/loss_function_test.py | 3 ++- tests/models/demucs_test.py | 6 +++--- tests/models/test_waveunet.py | 6 +++--- tests/utils_test.py | 7 ++++--- 4 files changed, 12 insertions(+), 10 deletions(-) diff --git a/tests/loss_function_test.py b/tests/loss_function_test.py index a4fdc62..cd60177 100644 --- a/tests/loss_function_test.py +++ b/tests/loss_function_test.py @@ -1,6 +1,7 @@ from asyncio import base_tasks -import torch + import pytest +import torch from enhancer.loss import mean_absolute_error, mean_squared_error diff --git a/tests/models/demucs_test.py b/tests/models/demucs_test.py index 6660888..1ea50c5 100644 --- a/tests/models/demucs_test.py +++ b/tests/models/demucs_test.py @@ -1,10 +1,10 @@ import pytest import torch -from enhancer import data -from enhancer.utils.config import Files -from enhancer.models import Demucs +from enhancer import data from enhancer.data.dataset import EnhancerDataset +from enhancer.models import Demucs +from enhancer.utils.config import Files @pytest.fixture diff --git a/tests/models/test_waveunet.py b/tests/models/test_waveunet.py index c83966b..798ed5d 100644 --- a/tests/models/test_waveunet.py +++ b/tests/models/test_waveunet.py @@ -1,10 +1,10 @@ import pytest import torch -from enhancer import data -from enhancer.utils.config import Files -from enhancer.models import WaveUnet +from enhancer import data from enhancer.data.dataset import EnhancerDataset +from enhancer.models import WaveUnet +from enhancer.utils.config import Files @pytest.fixture diff --git a/tests/utils_test.py b/tests/utils_test.py index 93a9094..1cc171a 100644 --- a/tests/utils_test.py +++ b/tests/utils_test.py @@ -1,11 +1,12 @@ from logging import root + +import numpy as np import pytest import torch -import numpy as np -from enhancer.utils.io import Audio -from enhancer.utils.config import Files from enhancer.data.fileprocessor import Fileprocessor +from enhancer.utils.config import Files +from enhancer.utils.io import Audio def test_io_channel():