def test_Matrix(): x = util.Matrix(("a", "b"), [[1, 2], [3, 4]]) assert x[0] == (1, 2) assert x[1] == (3, 4) assert x == ((1, 2), (3, 4)) assert repr(x) == "((1, 2), (3, 4))" with pytest.raises(TypeError): x[0][0] = 1
def test_Matrix(): m = util.Matrix(("a", "b")) m[:] = [[1, 2], [3, 4]] with PrintAssert("<Matrix ...>") as pr: m._repr_pretty_(pr, True) with PrintAssert(tx.matrix(m)) as pr: m._repr_pretty_(pr, False)
def test_Matrix(): m = util.Matrix(("a", "b")) m[:] = [[1, 1], [1, 4]] assert_equal(m, ((1, 1), (1, 4))) assert repr(m) == "[[1. 1.]\n [1. 4.]]" c = m.correlation() assert_allclose(c, ((1.0, 0.5), (0.5, 1.0))) assert m["a", "b"] == 1.0 assert m["a", 1] == 1.0 assert m[1, 1] == 4.0 assert_equal(m["b"], (1, 4)) m *= 2 assert_equal(m, ((2, 2), (2, 8))) assert_allclose(np.dot(m, (1, 1)), (4, 10)) with pytest.raises(TypeError): util.Matrix("ab") with pytest.raises(TypeError): util.Matrix(1)
def test_Matrix(): m = util.Matrix(("a", "b")) m[:] = [[1, 1], [1, 4]] assert_equal(m, ((1, 1), (1, 4))) assert repr(m) == "[[1. 1.]\n [1. 4.]]" c = m.correlation() assert_allclose(c, ((1.0, 0.5), (0.5, 1.0))) assert m["a", "b"] == 1.0 assert m["a", 1] == 1.0 assert m[1, 1] == 4.0 assert_equal(m["b"], (1, 4)) m *= 2 assert_equal(m, ((2, 2), (2, 8))) assert_allclose(np.dot(m, (1, 1)), (4, 10)) # matrix is always square m = util.Matrix(("a", "b", "c")) m[:] = np.arange(9).reshape((3, 3)) # [0 1 2 # 3 4 5 # 6 7 8] # m1 = m[:2] # assert_equal(m1, [[0, 1], [3, 4]]) m2 = m[[0, 2]] assert_equal(m2, [[0, 2], [6, 8]]) m3 = m[["a", "c"]] assert_equal(m3, [[0, 2], [6, 8]]) with pytest.raises(TypeError): util.Matrix("ab") with pytest.raises(TypeError): util.Matrix(1)