You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

14 lines
424 B

from sympy.tensor.tensor import TensorIndexType, tensor_indices, TensorHead
from sympy import I
def test_printing_TensMul():
R3 = TensorIndexType('R3', dim=3)
p, q = tensor_indices("p q", R3)
K = TensorHead("K", [R3])
assert repr(2*K(p)) == "2*K(p)"
assert repr(-K(p)) == "-K(p)"
assert repr(-2*K(p)*K(q)) == "-2*K(p)*K(q)"
assert repr(-I*K(p)) == "-I*K(p)"
assert repr(I*K(p)) == "I*K(p)"