Exemple #1
0
def test_linear_add_reduce():
    ops = LinearOperations()
    assert_almost_equal(ops.add_reduce(np.array([0,0,0])), 0)
    assert_almost_equal(ops.add_reduce(np.array([0,1,2])), 3)
    assert_almost_equal(ops.add_reduce(np.array([1,1,1])), 3)
    assert_almost_equal(ops.add_reduce(np.array([-1,0,1])), 0)
    assert_almost_equal(ops.add_reduce(np.array([2,0,-2])), 0)
    assert_almost_equal(ops.add_reduce(np.array([-1,-1,-1])), -3)
Exemple #2
0
def test_linear_mult():
    ops = LinearOperations()
    assert_almost_equal(ops.mult(0, 0), 0)
    assert_almost_equal(ops.mult(1, 0), 0)
    assert_almost_equal(ops.mult(0, 1), 0)
    assert_almost_equal(ops.mult(-1, 0), 0)
    assert_almost_equal(ops.mult(0, -1), 0)
    assert_almost_equal(ops.mult(1, -1), -1)
    assert_almost_equal(ops.mult(-1, 1), -1)
    assert_almost_equal(ops.mult(1, 1), 1)
    assert_almost_equal(ops.mult(-1, -1), 1)
    assert_almost_equal(ops.mult(2, 1), 2)
    assert_almost_equal(ops.mult(2, 2), 4)
    assert_almost_equal(ops.mult(2, -2), -4)
Exemple #3
0
def test_linear_add():
    ops = LinearOperations()
    assert_almost_equal(ops.add(0, 0), 0)
    assert_almost_equal(ops.add(1, 0), 1)
    assert_almost_equal(ops.add(0, 1), 1)
    assert_almost_equal(ops.add(-1, 0), -1)
    assert_almost_equal(ops.add(0, -1), -1)
    assert_almost_equal(ops.add(1, -1), 0)
    assert_almost_equal(ops.add(-1, 1), 0)
    assert_almost_equal(ops.add(1, 1), 2)
    assert_almost_equal(ops.add(-1, -1), -2)
Exemple #4
0
 def setup_class(self):
     self.ops = LinearOperations()
Exemple #5
0
class TestLinear(object):
    def setup_class(self):
        self.ops = LinearOperations()

    def test_add(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1])
        Z = np.array([0, 1, 1, -1, -1, 0, 0, 2, -2])
        for x, y, z in zip(X, Y, Z):
            assert self.ops.add(x, y) == pytest.approx(z)
        assert np.allclose(self.ops.add(X, Y), Z)

    def test_add_inplace(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1])
        Z = np.array([0, 1, 1, -1, -1, 0, 0, 2, -2])
        self.ops.add_inplace(X, Y)
        assert np.allclose(X, Z)

    def test_add_reduce(self):
        X = np.array([[0, 0, 0], [0, 1, 2], [1, 1, 1], [-1, 0, 1], [2, 0, -2],
                      [-1, -1, -1]])
        Y = np.array([0, 3, 3, 0, 0, -3])
        for x, y in zip(X, Y):
            assert self.ops.add_reduce(x) == pytest.approx(y)

    def test_mult(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1, 2, 2, 2])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1, 1, 2, -2])
        Z = np.array([0, 0, 0, 0, 0, -1, -1, 1, 1, 2, 4, -4])
        for x, y, z in zip(X, Y, Z):
            assert self.ops.mult(x, y) == pytest.approx(z)
        assert np.allclose(self.ops.mult(X, Y), Z)

    def test_mult_inplace(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1, 2, 2, 2])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1, 1, 2, -2])
        Z = np.array([0, 0, 0, 0, 0, -1, -1, 1, 1, 2, 4, -4])
        self.ops.mult_inplace(X, Y)
        assert np.allclose(X, Z)

    def test_invert(self):
        X = np.array([1, 2, -1, 10], dtype=float)
        Y = np.array([1, 1 / 2, -1, 1 / 10])
        for x, y in zip(X, Y):
            assert self.ops.invert(x) == pytest.approx(y)

    def test_mult_reduce(self):
        prods = [1, 2, 6, 24, 120]
        for i, p in enumerate(prods):
            assert self.ops.mult_reduce(np.arange(1,
                                                  i + 2)) == pytest.approx(p)

    def test_normalize(self):
        X = np.ones(3)
        Y = self.ops.normalize(X)
        Y_ = X / 3
        assert np.allclose(Y, Y_)
Exemple #6
0
 def setUp(self):
     self.ops = LinearOperations()
Exemple #7
0
class TestLinear(object):
    def setUp(self):
        self.ops = LinearOperations()

    def test_add(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1])
        Z = np.array([0, 1, 1, -1, -1, 0, 0, 2, -2])
        for x, y, z in zip(X, Y, Z):
            assert_almost_equal(self.ops.add(x, y), z)
        npt.assert_allclose(self.ops.add(X, Y), Z)

    def test_add_inplace(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1])
        Z = np.array([0, 1, 1, -1, -1, 0, 0, 2, -2])
        self.ops.add_inplace(X, Y)
        npt.assert_allclose(X, Z)

    def test_add_reduce(self):
        X = np.array([[0, 0, 0],
                      [0, 1, 2],
                      [1, 1, 1],
                      [-1, 0, 1],
                      [2, 0, -2],
                      [-1, -1, -1]])
        Y = np.array([0, 3, 3, 0, 0, -3])
        for x, y in zip(X, Y):
            assert_almost_equal(self.ops.add_reduce(x), y)

    def test_mult(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1, 2, 2, 2])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1, 1, 2, -2])
        Z = np.array([0, 0, 0, 0, 0, -1, -1, 1, 1, 2, 4, -4])
        for x, y, z in zip(X, Y, Z):
            assert_almost_equal(self.ops.mult(x, y), z)
        npt.assert_allclose(self.ops.mult(X, Y), Z)

    def test_mult_inplace(self):
        X = np.array([0, 1, 0, -1, 0, 1, -1, 1, -1, 2, 2, 2])
        Y = np.array([0, 0, 1, 0, -1, -1, 1, 1, -1, 1, 2, -2])
        Z = np.array([0, 0, 0, 0, 0, -1, -1, 1, 1, 2, 4, -4])
        self.ops.mult_inplace(X, Y)
        npt.assert_allclose(X, Z)

    def test_invert(self):
        X = np.array([1, 2, -1, 10], dtype=float)
        Y = np.array([1, 1/2, -1, 1/10])
        for x, y in zip(X, Y):
            assert_almost_equal(self.ops.invert(x), y)

    def test_mult_reduce(self):
        prods = [1, 2, 6, 24, 120]
        for i, p in enumerate(prods):
            assert_almost_equal(self.ops.mult_reduce(np.arange(1, i+2)), p)

    def test_normalize(self):
        X = np.ones(3)
        Y = self.ops.normalize(X)
        Y_ = X / 3
        npt.assert_allclose(Y, Y_)
Exemple #8
0
 def setup_class(self):
     self.ops = LinearOperations()
Exemple #9
0
def test_linear_invert():
    ops = LinearOperations()
    assert_almost_equal(ops.invert(1), 1)
    assert_almost_equal(ops.invert(2.0), 1/2)
    assert_almost_equal(ops.invert(-1), -1)
    assert_almost_equal(ops.invert(10.0), 1/10)