def test_less_that_one(self): factors = [0.1, 0.2, 0.3] scale = Scale(np.array(factors)) ones = np.ones(3) scaled_a = scale.scale(ones) testing.assert_array_equal(factors, scaled_a)
def test_basic(self): factors = [2, 3, 4] scale = Scale(np.array(factors)) ones = np.ones(3) scaled_a = scale.scale(ones) testing.assert_array_equal(factors, scaled_a)