--- a +++ b/tests/test_sampen.py @@ -0,0 +1,35 @@ +import numpy +import os +import unittest + +from pyeeg import samp_entropy + + +class SampEnTests(unittest.TestCase): + def test_sampen_against_predictable_sequence(self): + data = numpy.asarray([10, 20] * 2000) + self.assertAlmostEqual( + samp_entropy(data, 2, 0.2), + 0.0, + places=2 + ) + + def test_sampen_against_original_c_test_data(self): + """Use test data from + http://www.physionet.org/physiotools/sampen/c/sampentest.txt + """ + dir = os.path.dirname(__file__) + file_path = os.path.join(dir, './demo_data/sampentest.txt') + data = [] + with open(file_path, 'r') as file: + for row in file: + data.append(float(row.strip())) + + self.assertEqual( + samp_entropy(numpy.asarray(data), 2, 0.2), + 2.1233284920357112 + ) + + +if __name__ == '__main__': + unittest.main()