#!/usr/bin/env python

"""
test error computation functionality
"""

import numpy as np
import os
import unittest
from tvii import error

class ErrorTest(unittest.TestCase):

    def test_error_computation(self):
        """test error computation"""

        def square(x):
            return x*x

        x = [[1], [2], [3]]
        y = np.array([0, 3, 8])

        yhat, _err = error.error(square, x, y)

        assert list(yhat) == [1, 4, 9]
        assert list(_err) == [1, 1, 1]

if __name__ == '__main__':
    unittest.main()