def test_ipython(self):
alltext = []
class Pretty:
@staticmethod
def text(text):
alltext.append(text)
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
self.assertEqual(x._repr_html_(), r"\[kilogram\ meter^2/second\]")
self.assertEqual(
x._repr_latex_(),
r"$\frac{\mathrm{kilogram} \cdot "
r"\mathrm{meter}^{2}}{\mathrm{second}}$",
)
x._repr_pretty_(Pretty, False)
self.assertEqual("".join(alltext), "kilogram·meter²/second")
ureg.default_format = "~"
self.assertEqual(x._repr_html_(), r"\[kg\ m^2/s\]")
self.assertEqual(
x._repr_latex_(),
r"$\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}$")
alltext = []
x._repr_pretty_(Pretty, False)
self.assertEqual("".join(alltext), "kg·m²/s")
def test_ipython(self):
alltext = []
class Pretty:
@staticmethod
def text(text):
alltext.append(text)
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
assert x._repr_html_() == "kilogram meter<sup>2</sup>/second"
assert (
x._repr_latex_() == r"$\frac{\mathrm{kilogram} \cdot "
r"\mathrm{meter}^{2}}{\mathrm{second}}$"
)
x._repr_pretty_(Pretty, False)
assert "".join(alltext) == "kilogram·meter²/second"
ureg.default_format = "~"
assert x._repr_html_() == "kg m<sup>2</sup>/s"
assert (
x._repr_latex_() == r"$\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}$"
)
alltext = []
x._repr_pretty_(Pretty, False)
assert "".join(alltext) == "kg·m²/s"
def test_default_format(self):
ureg = UnitRegistry()
q = ureg.meter
s1 = '{0}'.format(q)
s2 = '{0:~}'.format(q)
ureg.default_format = '~'
s3 = '{0}'.format(q)
self.assertEqual(s2, s3)
self.assertNotEqual(s1, s3)
self.assertEqual(ureg.default_format, '~')
def test_default_format(self):
ureg = UnitRegistry()
q = ureg.meter
s1 = f"{q}"
s2 = f"{q:~}"
ureg.default_format = "~"
s3 = f"{q}"
self.assertEqual(s2, s3)
self.assertNotEqual(s1, s3)
self.assertEqual(ureg.default_format, "~")
def test_default_format(self):
ureg = UnitRegistry()
q = ureg.meter
s1 = '{0}'.format(q)
s2 = '{0:~}'.format(q)
ureg.default_format = '~'
s3 = '{0}'.format(q)
self.assertEqual(s2, s3)
self.assertNotEqual(s1, s3)
self.assertEqual(ureg.default_format, '~')
def test_default_format(self):
ureg = UnitRegistry()
q = ureg.meter
s1 = f"{q}"
s2 = f"{q:~}"
ureg.default_format = "~"
s3 = f"{q}"
assert s2 == s3
assert s1 != s3
assert ureg.default_format == "~"
def test_unit_default_formatting(self):
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
for spec, result in (('L', r'\frac{\mathrm{kilogram} \cdot \mathrm{meter}^{2}}{\mathrm{second}}'),
('P', 'kilogram·meter²/second'),
('H', 'kilogram meter<sup>2</sup>/second'),
('C', 'kilogram*meter**2/second'),
('~', 'kg * m ** 2 / s'),
('L~', r'\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}'),
('P~', 'kg·m²/s'),
('H~', 'kg m<sup>2</sup>/s'),
('C~', 'kg*m**2/s'),
):
ureg.default_format = spec
self.assertEqual('{0}'.format(x), result,
'Failed for {0}, {1}'.format(spec, result))
def test_unit_default_formatting(self):
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
for spec, result in (('L', r'\frac{\mathrm{kilogram} \cdot \mathrm{meter}^{2}}{\mathrm{second}}'),
('P', 'kilogram·meter²/second'),
('H', 'kilogram meter<sup>2</sup>/second'),
('C', 'kilogram*meter**2/second'),
('~', 'kg * m ** 2 / s'),
('L~', r'\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}'),
('P~', 'kg·m²/s'),
('H~', 'kg m<sup>2</sup>/s'),
('C~', 'kg*m**2/s'),
):
ureg.default_format = spec
self.assertEqual('{0}'.format(x), result,
'Failed for {0}, {1}'.format(spec, result))
def test_unit_formatting_snake_case(self):
# Test that snake_case units are escaped where appropriate
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(oil_barrel=1))
for spec, result in (
("L", r"\mathrm{oil\_barrel}"),
("P", "oil_barrel"),
("H", r"\[oil\_barrel\]"),
("C", "oil_barrel"),
("~", "oil_bbl"),
("L~", r"\mathrm{oil\_bbl}"),
("P~", "oil_bbl"),
("H~", r"\[oil\_bbl\]"),
("C~", "oil_bbl"),
):
with self.subTest(spec):
ureg.default_format = spec
self.assertEqual(f"{x}", result, f"Failed for {spec}, {result}")
def test_unit_default_formatting(self):
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
for spec, result in (
(
"L",
r"\frac{\mathrm{kilogram} \cdot \mathrm{meter}^{2}}{\mathrm{second}}",
),
("P", "kilogram·meter²/second"),
("H", r"\[kilogram\ meter^2/second\]"),
("C", "kilogram*meter**2/second"),
("~", "kg * m ** 2 / s"),
("L~", r"\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}"),
("P~", "kg·m²/s"),
("H~", r"\[kg\ m^2/s\]"),
("C~", "kg*m**2/s"),
):
with self.subTest(spec):
ureg.default_format = spec
self.assertEqual(f"{x}", result, f"Failed for {spec}, {result}")
def test_unit_default_formatting(self, subtests):
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
for spec, result in (
(
"L",
r"\frac{\mathrm{kilogram} \cdot \mathrm{meter}^{2}}{\mathrm{second}}",
),
("P", "kilogram·meter²/second"),
("H", "kilogram meter<sup>2</sup>/second"),
("C", "kilogram*meter**2/second"),
("~", "kg * m ** 2 / s"),
("L~", r"\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}"),
("P~", "kg·m²/s"),
("H~", "kg m<sup>2</sup>/s"),
("C~", "kg*m**2/s"),
):
with subtests.test(spec):
ureg.default_format = spec
assert f"{x}" == result, f"Failed for {spec}, {result}"
def test_ipython(self):
alltext = []
class Pretty(object):
@staticmethod
def text(text):
alltext.append(text)
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
self.assertEqual(x._repr_html_(), "kilogram meter<sup>2</sup>/second")
self.assertEqual(x._repr_latex_(), r'$\frac{\mathrm{kilogram} \cdot '
r'\mathrm{meter}^{2}}{\mathrm{second}}$')
x._repr_pretty_(Pretty, False)
self.assertEqual("".join(alltext), "kilogram·meter²/second")
ureg.default_format = "~"
self.assertEqual(x._repr_html_(), "kg m<sup>2</sup>/s")
self.assertEqual(x._repr_latex_(),
r'$\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}$')
alltext = []
x._repr_pretty_(Pretty, False)
self.assertEqual("".join(alltext), "kg·m²/s")
def test_ipython(self):
alltext = []
class Pretty(object):
@staticmethod
def text(text):
alltext.append(text)
ureg = UnitRegistry()
x = ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
self.assertEqual(x._repr_html_(), "kilogram meter<sup>2</sup>/second")
self.assertEqual(x._repr_latex_(), r'$\frac{\mathrm{kilogram} \cdot '
r'\mathrm{meter}^{2}}{\mathrm{second}}$')
x._repr_pretty_(Pretty, False)
self.assertEqual("".join(alltext), "kilogram·meter²/second")
ureg.default_format = "~"
self.assertEqual(x._repr_html_(), "kg m<sup>2</sup>/s")
self.assertEqual(x._repr_latex_(),
r'$\frac{\mathrm{kg} \cdot \mathrm{m}^{2}}{\mathrm{s}}$')
alltext = []
x._repr_pretty_(Pretty, False)
self.assertEqual("".join(alltext), "kg·m²/s")
