test_annotation: convert to a standard unittest.TestCase.

Make the test_annotation class a subclass of unittest.TestCase,
allowing it to use standard unit testing utility methods, as well as
setup and teardown functions.  (nosetests will run "test" class
methods automatically even if they are not subclasses of TestCase, but
unittest won't.)  Rename the class to TestAnnotation for consistency.

Make the module executable (invoke unittest.main()) so it can be
invoked simply using 'python3 -m tests.test_annotation'.

Ensure that temporary files created by the annotation tests will be
correctly cleaned up by TestAnnotation.tearDownClass() rather than by
the unrelated TestRecord.tearDownClass().  (Presumably this only
happened to work previously because "test_record" comes alphabetically
after "test_annotation".)

Author: Benjamin Moody

tests/test_annotation.py

@@ -1,10 +1,13 @@
+import os
 import re
+import unittest
 
 import numpy as np
 
 import wfdb
 
-class test_annotation():
+
+class TestAnnotation(unittest.TestCase):
     """
     Testing read and write of WFDB annotations, including Physionet
     streaming.
@@ -183,3 +186,18 @@ def test_3(self):
         assert (comp == [True] * 6)
         assert annotation.__eq__(pn_annotation)
         assert annotation.__eq__(write_annotation)
+
+    @classmethod
+    def tearDownClass(cls):
+        writefiles = [
+            '100.atr',
+            '1003.atr',
+            '12726.anI',
+        ]
+        for file in writefiles:
+            if os.path.isfile(file):
+                os.remove(file)
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_record.py

@@ -521,9 +521,9 @@ def test_header_with_non_utf8(self):
     @classmethod
     def tearDownClass(cls):
         "Clean up written files"
-        writefiles = ['03700181.dat','03700181.hea','100.atr','100.dat',
-                      '100.hea','1003.atr','100_3chan.dat','100_3chan.hea',
-                      '12726.anI','a103l.hea','a103l.mat','s0010_re.dat',
+        writefiles = ['03700181.dat','03700181.hea','100.dat',
+                      '100.hea','100_3chan.dat','100_3chan.hea',
+                      'a103l.hea','a103l.mat','s0010_re.dat',
                       's0010_re.hea','s0010_re.xyz','test01_00s.dat',
                       'test01_00s.hea','test01_00s_skewframe.hea',
                       'n8_evoked_raw_95_F1_R9.dat', 'n8_evoked_raw_95_F1_R9.hea']