summaryrefslogtreecommitdiff
path: root/pint/testsuite/test_infer_base_unit.py
diff options
context:
space:
mode:
Diffstat (limited to 'pint/testsuite/test_infer_base_unit.py')
-rw-r--r--pint/testsuite/test_infer_base_unit.py18
1 files changed, 8 insertions, 10 deletions
diff --git a/pint/testsuite/test_infer_base_unit.py b/pint/testsuite/test_infer_base_unit.py
index dc3bd82..54a4720 100644
--- a/pint/testsuite/test_infer_base_unit.py
+++ b/pint/testsuite/test_infer_base_unit.py
@@ -1,32 +1,30 @@
from pint import Quantity as Q
-from pint.testsuite import QuantityTestCase
+from pint.testsuite import helpers
from pint.util import infer_base_unit
-class TestInferBaseUnit(QuantityTestCase):
+class TestInferBaseUnit:
def test_infer_base_unit(self):
from pint.util import infer_base_unit
- self.assertEqual(
- infer_base_unit(Q(1, "millimeter * nanometer")), Q(1, "meter**2").units
- )
+ assert infer_base_unit(Q(1, "millimeter * nanometer")) == Q(1, "meter**2").units
def test_units_adding_to_zero(self):
- self.assertEqual(infer_base_unit(Q(1, "m * mm / m / um * s")), Q(1, "s").units)
+ assert infer_base_unit(Q(1, "m * mm / m / um * s")) == Q(1, "s").units
def test_to_compact(self):
r = Q(1000000000, "m") * Q(1, "mm") / Q(1, "s") / Q(1, "ms")
compact_r = r.to_compact()
expected = Q(1000.0, "kilometer**2 / second**2")
- self.assertQuantityAlmostEqual(compact_r, expected)
+ helpers.assert_quantity_almost_equal(compact_r, expected)
r = (Q(1, "m") * Q(1, "mm") / Q(1, "m") / Q(2, "um") * Q(2, "s")).to_compact()
- self.assertQuantityAlmostEqual(r, Q(1000, "s"))
+ helpers.assert_quantity_almost_equal(r, Q(1000, "s"))
def test_volts(self):
from pint.util import infer_base_unit
r = Q(1, "V") * Q(1, "mV") / Q(1, "kV")
b = infer_base_unit(r)
- self.assertEqual(b, Q(1, "V").units)
- self.assertQuantityAlmostEqual(r, Q(1, "uV"))
+ assert b == Q(1, "V").units
+ helpers.assert_quantity_almost_equal(r, Q(1, "uV"))