#!/usr/bin/env python

import unittest
from interp.baker import get_phis, qlinear


class Test(unittest.TestCase):
    def setUp(self):
        self.X = [0.0, 0.0, 0.0]
        self.r = [
            [0.0, 0.0, 1.0],
            [0.94280904333606508, 0.0, -0.3333333283722672],
            [-0.47140452166803232, 0.81649658244673617, -0.3333333283722672],
            [-0.47140452166803298, -0.81649658244673584, -0.3333333283722672],
        ]
        self.q = [0.0, 0.0, 0.0, 4]

    def testGetPhis(self):
        result = get_phis(self.X, self.r)
        right_answer = [0.25, 0.25, 0.25, 0.25]

        for a, b in zip(result, right_answer):
            self.assertAlmostEqual(a, b)

    def testQlinear(self):
        phi, result = qlinear(self.X, self.r, self.q)
        result = result
        right_answer = 1.0
        self.assertAlmostEqual(result, right_answer)


if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(Test)
    unittest.TextTestRunner(verbosity=2).run(suite)