def testOneColumn(self): inarray = [[0.], [1.], [2.]] n = _norms(inarray) bool1 = n[0].tolist() == [[0., 2.]] bool2 = n[1].tolist() == [[0.5, 0.]] bool3 = n[2].tolist() == [[2., 0.]] self.assert_(bool1 and bool2 and bool3)
def testNormalCase(self): inarray = [[0.,0.], [0.,1.], [1.,0.], [1.,1.]] n = _norms(inarray, lower=0.15, upper=0.85) self.assertEqual(n[0].tolist(), [[0., 1.], [0, 1.]]) self.assertEqual(n[1].tolist(), [[0.7, 0.15], [0.7, 0.15]]) self.assertAlmostEqual(n[2][0,0], 1.42857143, 8) self.assertAlmostEqual(n[2][0,1], -0.21428571, 8)
def testEmpty(self): inarray = [[], []] n = _norms(inarray) for i in xrange(len(n)): self.assertEqual(n[i].tolist(), [])