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(), [])
