add tests

philipperemy · Oct 4, 2022 · 3a327f8 · 3a327f8
1 parent cd8fda7
commit 3a327f8
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 11 deletions.
diff --git a/tests/__init__.py → deep_speaker/tests/__init__.py b/tests/__init__.py → deep_speaker/tests/__init__.py
diff --git a/tests/batcher_t1.py → deep_speaker/tests/batcher_t1.py b/tests/batcher_t1.py → deep_speaker/tests/batcher_t1.py
@@ -2,12 +2,12 @@
 
 import numpy as np
 
-import triplet_loss
-from batcher import KerasFormatConverter, TripletBatcherSelectHardNegatives, TripletBatcher
-from constants import NUM_FBANKS, NUM_FRAMES, CHECKPOINTS_TRIPLET_DIR, CHECKPOINTS_SOFTMAX_DIR, BATCH_SIZE
-from conv_models import DeepSpeakerModel
-from triplet_loss import deep_speaker_loss
-from utils import load_best_checkpoint
+from deep_speaker import triplet_loss
+from deep_speaker.batcher import KerasFormatConverter, TripletBatcherSelectHardNegatives, TripletBatcher
+from deep_speaker.constants import NUM_FBANKS, NUM_FRAMES, CHECKPOINTS_TRIPLET_DIR, CHECKPOINTS_SOFTMAX_DIR, BATCH_SIZE
+from deep_speaker.conv_models import DeepSpeakerModel
+from deep_speaker.triplet_loss import deep_speaker_loss
+from deep_speaker.utils import load_best_checkpoint
 
 
 def predict(x):

diff --git a/tests/batcher_t2.py → deep_speaker/tests/batcher_t2.py b/tests/batcher_t2.py → deep_speaker/tests/batcher_t2.py
@@ -1,9 +1,9 @@
 import numpy as np
 
-from batcher import LazyTripletBatcher
-from constants import NUM_FBANKS, NUM_FRAMES
-from conv_models import DeepSpeakerModel
-from triplet_loss import deep_speaker_loss
+from deep_speaker.batcher import LazyTripletBatcher
+from deep_speaker.constants import NUM_FBANKS, NUM_FRAMES
+from deep_speaker.conv_models import DeepSpeakerModel
+from deep_speaker.triplet_loss import deep_speaker_loss
 
 
 def main2():

diff --git a/tests/triplet_loss_test.py → deep_speaker/tests/triplet_loss_test.py b/tests/triplet_loss_test.py → deep_speaker/tests/triplet_loss_test.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from triplet_loss import deep_speaker_loss
+from deep_speaker.triplet_loss import deep_speaker_loss
 
 
 def opposite_positive_equal_negative_batch():

diff --git a/tox.ini b/tox.ini
@@ -16,5 +16,6 @@ commands = pylint --disable=R,C,W,E1136 deep_speaker
            flake8 deep_speaker --count --exclude=michel,tests --max-line-length 127 --statistics
            python example.py
            python example_pcm.py
+           pytest tests
 passenv = *
 install_command = pip install {packages}