def test_integration():
import scipy.integrate as integrate
m = u.phval(1, "m")
s = u.phval(1, "s")
# Velocity in m/s
def v(t):
a = 10 * m / (s * s)
return a * t
integral = u.phval(integrate.quad(v, 0 * s, 10 * s)[0], "1/eV")
print(integral.str("m"))
def test_plot():
import matplotlib.pyplot as plt
x_units = 'kg'
y_units = 's'
x = u.phval(np.linspace(0.1, 10, 100), x_units)
y = u.phval(np.linspace(0.1, 10, 100), y_units)
plt.plot(x.val_u(x_units), y.val_u(y_units))
plt.xlabel(x_units)
plt.ylabel(y_units)
def test_comparison_functions():
comparison_func = {
np.greater, np.greater_equal, np.less, np.less_equal, np.not_equal,
np.equal
}
ev1 = u.phval(1, 1)
ev2 = u.phval(10, 1)
noev = u.phval(1, -1)
zero = u.phval(0., -1)
for f in comparison_func:
f(ev1, ev2)
f(zero, ev1)
with raises(u.UnitError):
f(ev1, noev)
def test_operations(units):
x = u.phval(1, units[0])
y = u.phval(1, units[1])
print(x, units[0])
print(y, units[1])
z = x + y
assert float(z) == float(x) + float(y)
assert z.units == x.units == y.units
z = x - y
assert float(z) == float(x) - float(y)
assert z.units == x.units == y.units
z = x / y
assert float(z) == float(x) / float(y)
assert z.units == x.units - y.units
z = x * y
assert float(z) == float(x) * float(y)
assert z.units == x.units + y.units
def test_unitless_functions():
kg = u.phval(1, "kg")
unitless_fun = [
np.sin, np.cos, np.exp, np.arcsin, np.arccos, np.tan, np.arctan,
np.log, np.log10
]
with raises(u.UnitError):
for f in unitless_fun:
f(kg)
def test_integration_odeint():
from scipy.integrate import odeint
N = u.phval(1, "N")
kg = u.phval(1, "kg")
m = u.phval(1, "m")
s = u.phval(1, "s")
# function that returns dy/dt
def model(y, t):
k = 0.3 * N / m
dydt = -k * y
return dydt
# initial condition
y0 = 5 * m
# time points
t = np.linspace(0, 20) * s
#t.astype(np.float64, casting="unsafe")
# solve ODE
y = odeint(model, y0, t.values)
def test_creates_class(units):
u.phval(1., units)