def test_partition_of_unity(self): bs = BSpline() B = bs.spline_matrix(self.z) unity_vector = np.sum(B, axis=1) print "Matrix B" print B print "Partition of unity check" print unity_vector N, M = np.shape(B) print "checking unity_vector ..." self.assertEqual(N, sum(unity_vector)) pass