Spaces:
Sleeping
Sleeping
| from sympy.physics.units.systems.si import dimsys_SI | |
| from sympy.core.numbers import pi | |
| from sympy.core.singleton import S | |
| from sympy.core.symbol import Symbol | |
| from sympy.functions.elementary.complexes import Abs | |
| from sympy.functions.elementary.exponential import log | |
| from sympy.functions.elementary.miscellaneous import sqrt | |
| from sympy.functions.elementary.trigonometric import (acos, atan2, cos) | |
| from sympy.physics.units.dimensions import Dimension | |
| from sympy.physics.units.definitions.dimension_definitions import ( | |
| length, time, mass, force, pressure, angle | |
| ) | |
| from sympy.physics.units import foot | |
| from sympy.testing.pytest import raises | |
| def test_Dimension_definition(): | |
| assert dimsys_SI.get_dimensional_dependencies(length) == {length: 1} | |
| assert length.name == Symbol("length") | |
| assert length.symbol == Symbol("L") | |
| halflength = sqrt(length) | |
| assert dimsys_SI.get_dimensional_dependencies(halflength) == {length: S.Half} | |
| def test_Dimension_error_definition(): | |
| # tuple with more or less than two entries | |
| raises(TypeError, lambda: Dimension(("length", 1, 2))) | |
| raises(TypeError, lambda: Dimension(["length"])) | |
| # non-number power | |
| raises(TypeError, lambda: Dimension({"length": "a"})) | |
| # non-number with named argument | |
| raises(TypeError, lambda: Dimension({"length": (1, 2)})) | |
| # symbol should by Symbol or str | |
| raises(AssertionError, lambda: Dimension("length", symbol=1)) | |
| def test_str(): | |
| assert str(Dimension("length")) == "Dimension(length)" | |
| assert str(Dimension("length", "L")) == "Dimension(length, L)" | |
| def test_Dimension_properties(): | |
| assert dimsys_SI.is_dimensionless(length) is False | |
| assert dimsys_SI.is_dimensionless(length/length) is True | |
| assert dimsys_SI.is_dimensionless(Dimension("undefined")) is False | |
| assert length.has_integer_powers(dimsys_SI) is True | |
| assert (length**(-1)).has_integer_powers(dimsys_SI) is True | |
| assert (length**1.5).has_integer_powers(dimsys_SI) is False | |
| def test_Dimension_add_sub(): | |
| assert length + length == length | |
| assert length - length == length | |
| assert -length == length | |
| raises(TypeError, lambda: length + foot) | |
| raises(TypeError, lambda: foot + length) | |
| raises(TypeError, lambda: length - foot) | |
| raises(TypeError, lambda: foot - length) | |
| # issue 14547 - only raise error for dimensional args; allow | |
| # others to pass | |
| x = Symbol('x') | |
| e = length + x | |
| assert e == x + length and e.is_Add and set(e.args) == {length, x} | |
| e = length + 1 | |
| assert e == 1 + length == 1 - length and e.is_Add and set(e.args) == {length, 1} | |
| assert dimsys_SI.get_dimensional_dependencies(mass * length / time**2 + force) == \ | |
| {length: 1, mass: 1, time: -2} | |
| assert dimsys_SI.get_dimensional_dependencies(mass * length / time**2 + force - | |
| pressure * length**2) == \ | |
| {length: 1, mass: 1, time: -2} | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(mass * length / time**2 + pressure)) | |
| def test_Dimension_mul_div_exp(): | |
| assert 2*length == length*2 == length/2 == length | |
| assert 2/length == 1/length | |
| x = Symbol('x') | |
| m = x*length | |
| assert m == length*x and m.is_Mul and set(m.args) == {x, length} | |
| d = x/length | |
| assert d == x*length**-1 and d.is_Mul and set(d.args) == {x, 1/length} | |
| d = length/x | |
| assert d == length*x**-1 and d.is_Mul and set(d.args) == {1/x, length} | |
| velo = length / time | |
| assert (length * length) == length ** 2 | |
| assert dimsys_SI.get_dimensional_dependencies(length * length) == {length: 2} | |
| assert dimsys_SI.get_dimensional_dependencies(length ** 2) == {length: 2} | |
| assert dimsys_SI.get_dimensional_dependencies(length * time) == {length: 1, time: 1} | |
| assert dimsys_SI.get_dimensional_dependencies(velo) == {length: 1, time: -1} | |
| assert dimsys_SI.get_dimensional_dependencies(velo ** 2) == {length: 2, time: -2} | |
| assert dimsys_SI.get_dimensional_dependencies(length / length) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(velo / length * time) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(length ** -1) == {length: -1} | |
| assert dimsys_SI.get_dimensional_dependencies(velo ** -1.5) == {length: -1.5, time: 1.5} | |
| length_a = length**"a" | |
| assert dimsys_SI.get_dimensional_dependencies(length_a) == {length: Symbol("a")} | |
| assert dimsys_SI.get_dimensional_dependencies(length**pi) == {length: pi} | |
| assert dimsys_SI.get_dimensional_dependencies(length**(length/length)) == {length: Dimension(1)} | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(length**length)) | |
| assert length != 1 | |
| assert length / length != 1 | |
| length_0 = length ** 0 | |
| assert dimsys_SI.get_dimensional_dependencies(length_0) == {} | |
| # issue 18738 | |
| a = Symbol('a') | |
| b = Symbol('b') | |
| c = sqrt(a**2 + b**2) | |
| c_dim = c.subs({a: length, b: length}) | |
| assert dimsys_SI.equivalent_dims(c_dim, length) | |
| def test_Dimension_functions(): | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(cos(length))) | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(acos(angle))) | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(atan2(length, time))) | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(log(length))) | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(log(100, length))) | |
| raises(TypeError, lambda: dimsys_SI.get_dimensional_dependencies(log(length, 10))) | |
| assert dimsys_SI.get_dimensional_dependencies(pi) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(cos(1)) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(cos(angle)) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(atan2(length, length)) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(log(length / length, length / length)) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(Abs(length)) == {length: 1} | |
| assert dimsys_SI.get_dimensional_dependencies(Abs(length / length)) == {} | |
| assert dimsys_SI.get_dimensional_dependencies(sqrt(-1)) == {} | |