test_printing.py 424 B

12345678910111213
  1. from sympy.tensor.tensor import TensorIndexType, tensor_indices, TensorHead
  2. from sympy import I
  3. def test_printing_TensMul():
  4. R3 = TensorIndexType('R3', dim=3)
  5. p, q = tensor_indices("p q", R3)
  6. K = TensorHead("K", [R3])
  7. assert repr(2*K(p)) == "2*K(p)"
  8. assert repr(-K(p)) == "-K(p)"
  9. assert repr(-2*K(p)*K(q)) == "-2*K(p)*K(q)"
  10. assert repr(-I*K(p)) == "-I*K(p)"
  11. assert repr(I*K(p)) == "I*K(p)"