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)