1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#!/usr/bin/env python
# Licensed under a 3-clause BSD style license, see LICENSE.
"""
Tests for the hepunits.pint module.
"""
import pint
import pytest
from pytest import approx
import hepunits
from hepunits.pint import from_clhep, to_clhep
def test_pint_constants():
ureg = pint.UnitRegistry()
# These three constants set the relationship between mass, length, time, and charge
# TODO: check all 7 SI defining constants (hence also supporting more unit conversions)
pint_c = to_clhep(1 * ureg.speed_of_light)
assert pint_c == approx(hepunits.c_light, rel=1e-15)
pint_h = to_clhep(1 * ureg.planck_constant)
assert pint_h == approx(hepunits.h_Planck, rel=1e-15)
pint_e = to_clhep(1 * ureg.elementary_charge)
assert pint_e == approx(1.0, rel=1e-15)
def test_pint_roundtrip():
ureg = pint.UnitRegistry()
assert to_clhep(3 * ureg.mm) == approx(3.0)
assert to_clhep(3 * ureg.cm) == approx(30.0)
assert to_clhep(2 * ureg.ohm) == approx(2.0 * hepunits.ohm)
assert to_clhep(ureg.coulomb) == approx(hepunits.coulomb)
assert from_clhep(hepunits.c_light, ureg.meter / ureg.second).m == approx(
(1.0 * ureg.c).to(ureg.meter / ureg.second).m, rel=1e-15
)
assert from_clhep(hepunits.tesla, ureg.tesla).m == approx(
(1 * ureg.tesla).m, rel=1e-15
)
def test_unsupported_dimension():
ureg = pint.UnitRegistry()
with pytest.raises(ValueError, match="Unsupported dimension"):
to_clhep(1 * ureg.kelvin)
def test_consistent_registry():
ureg = pint.UnitRegistry()
a = 1 * hepunits.mm
b = 3 * ureg.nanosecond
assert a * to_clhep(b) == 3 * hepunits.mm * hepunits.nanosecond
assert from_clhep(a, ureg.mm) * b == 3 * ureg.mm * ureg.nanosecond
|
