tvii: tests/test_error.py comparison

add error computation + test functions

comparison

equal deleted inserted replaced

-:0bb36ae047c3
+:351fc97bb996
+#!/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()

Mercurial > hg > tvii