def test_normalize_with_axis_3D(self): A = np.random.rand(10, 4, 5) + 1.0 for axis in range(3): Anorm = gmm.normalize(A, axis) self.assertTrue(np.all(np.allclose(Anorm.sum(axis), 1.0)))
def test_normalize_3D(): A = np.random.rand(2, 2, 2) + 1.0 for axis in range(3): Anorm = gmm.normalize(A, axis) assert np.all(np.allclose(Anorm.sum(axis), 1.0))
def test_normalize_2D(self): A = np.random.rand(10, 4) + 1.0 Anorm = gmm.normalize(A) self.assertAlmostEqual(Anorm.sum(), 1.0)