shithub: aubio

Info  •  Files  •  Log  •  Branches

[tests] move unittest import to main

--- a/python/tests/test_slicing.py

+++ b/python/tests/test_slicing.py

@@ -1,6 +1,5 @@

 #! /usr/bin/env python

-from unittest import main

 from numpy.testing import TestCase, assert_equal

 from aubio import slice_source_at_stamps

 from .utils import count_files_in_directory, get_default_test_sound

@@ -167,4 +166,5 @@

         shutil.rmtree(self.output_dir)

 if __name__ == '__main__':

+    from unittest import main

     main()

--- a/python/tests/test_specdesc.py

+++ b/python/tests/test_specdesc.py

@@ -1,6 +1,5 @@

 #! /usr/bin/env python

-from unittest import main

 from numpy.testing import TestCase, assert_equal, assert_almost_equal

 from numpy import random, arange, log, zeros

 from aubio import specdesc, cvec, float_type

@@ -229,4 +228,5 @@

             specdesc("unknown", 512)

 if __name__ == '__main__':

+    from unittest import main

     main()

--- a/python/tests/test_zero_crossing_rate.py

+++ b/python/tests/test_zero_crossing_rate.py

@@ -1,6 +1,5 @@

 #! /usr/bin/env python

-from unittest import main

 from numpy.testing import TestCase

 from aubio import fvec, zero_crossing_rate

@@ -43,4 +42,5 @@

         self.assertEqual(2./buf_size, zero_crossing_rate(self.vector))

 if __name__ == '__main__':

+    from unittest import main

     main()