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()