# coding=UTF-8
from __future__ import print_function, absolute_import, division
import six
import unittest
from firanka.series import DiscreteSeries, FunctionBasedSeries, Range, ModuloSeries


class TestDiscreteSeries(unittest.TestCase):

    def test_base(self):
        s = DiscreteSeries([[0,0], [1,1], [2,2]])

        self.assertEqual(s[0], 0)
        self.assertEqual(s[0.5], 0)
        self.assertEqual(s[1], 1)

        self.assertRaises(ValueError, lambda: s[-1])
        self.assertRaises(ValueError, lambda: s[2.5])

        s = DiscreteSeries([[0,0], [1,1], [2,2]], domain=Range(0,3,True,True))
        self.assertEqual(s[0], 0)
        self.assertEqual(s[0.5], 0)
        self.assertEqual(s[1], 1)

        self.assertRaises(ValueError, lambda: s[-1])
        self.assertEqual(s[2.5], 2)

    def test_translation(self):
        s = DiscreteSeries([[0,0], [1,1], [2,2]]).translate(3)

        self.assertEqual(s[3], 0)
        self.assertEqual(s[3.5], 0)
        self.assertEqual(s[4], 1)

    def test_slice(self):
        series = DiscreteSeries([[0, 0], [1, 1], [2, 2]])

        sp = series[0.5:1.5]

        self.assertEqual(sp[0.5], 0)
        self.assertEqual(sp[1.5], 1)
        self.assertRaises(ValueError, lambda: sp[0])
        self.assertRaises(ValueError, lambda: sp[2])
        self.assertEqual(sp.domain.start, 0.5)
        self.assertEqual(sp.domain.stop, 1.5)

    def test_eval(self):
        sa = DiscreteSeries([[0, 0], [1, 1], [2, 2]])
        sb = DiscreteSeries([[0, 1], [1, 2], [2, 3]])

        sc = sa.join_discrete(sb, lambda a, b: a+b)
        self.assertIsInstance(sc, DiscreteSeries)
        self.assertEqual(sc.eval_points([0,1,2]), [1,3,5])
        self.assertEqual(sc.data, [(0,1),(1,3),(2,5)])

    def test_eval2(self):
        sa = DiscreteSeries([[0, 0], [1, 1], [2, 2]])
        sb = FunctionBasedSeries(lambda x: x, '<0;2>')

        sc = sa.join_discrete(sb, lambda a, b: a+b)
        self.assertEqual(sc.eval_points([0,1,2]), [0,2,4])

        self.assertIsInstance(sc, DiscreteSeries)
        self.assertEqual(sc.data, [(0,0),(1,2),(2,4)])

    def test_apply(self):
        sa = DiscreteSeries([[0, 0], [1, 1], [2, 2]]).apply(lambda x: x+1)
        self.assertEquals(sa.data, [(0,1),(1,2),(2,3)])

    def test_eval3(self):
        sa = FunctionBasedSeries(lambda x: x**2, '<-10;10)')
        sb = FunctionBasedSeries(lambda x: x, '<0;2)')

        sc = sa.join(sb, lambda a, b: a*b)

        PTS = [0,1,1.9]
        EPTS = [x*x**2 for x in PTS]

        self.assertEqual(sc.eval_points(PTS), EPTS)
        self.assertTrue(Range('<0;2)') in sc.domain)

    def test_discretize(self):
        PTS = [0,1,2,3,4,5]
        sa = FunctionBasedSeries(lambda x: x**2, '<-10;10)').discretize(PTS, '(-1;6)')
        self.assertIsInstance(sa, DiscreteSeries)
        self.assertEqual(sa.data, [(i, i**2) for i in PTS])

        sa = FunctionBasedSeries(lambda x: x**2, '<-10;10)').discretize(PTS)
        self.assertIsInstance(sa, DiscreteSeries)
        self.assertEqual(sa.data, [(i, i**2) for i in PTS])

class TestFunctionBasedSeries(unittest.TestCase):
    def test_slice(self):
        series = FunctionBasedSeries(lambda x: x, '<0;2>')

        sp = series[0.5:1.5]

        self.assertEqual(sp[0.5], 0.5)
        self.assertEqual(sp[1.5], 1.5)
        self.assertRaises(ValueError, lambda: sp[0])
        self.assertRaises(ValueError, lambda: sp[2])
        self.assertEqual(sp.domain.start, 0.5)
        self.assertEqual(sp.domain.stop, 1.5)


class TestModuloSeries(unittest.TestCase):
    def test_base(self):
        series = ModuloSeries(DiscreteSeries([(0,1),(1,2),(2,3)], '<0;3)'))

        self.assertEquals(series[3], 1)
        self.assertEquals(series[4], 2)
        self.assertEquals(series[5], 3)
        self.assertEquals(series[-1], 3)

    def test_comp_discrete(self):
        ser1 = ModuloSeries(FunctionBasedSeries(lambda x: x**2, '<0;3)'))
        ser2 = FunctionBasedSeries(lambda x: x, '<0;3)')

        ser3 = ser1.join(ser2, lambda x, y: x*y)