add tests for the IPython _repr_ methods

_repr_html_ and _repr_latex_ were untested

2 files changed, 16 insertions, 0 deletions

diff --git a/pint/testsuite/test_quantity.py b/pint/testsuite/test_quantity.py
index 29ecfea..a05387b 100644
--- a/pint/testsuite/test_quantity.py
+++ b/pint/testsuite/test_quantity.py
@@ -166,6 +166,15 @@ class TestQuantity(QuantityTestCase):
             ureg.default_format = spec
             self.assertEqual('{0}'.format(x), result)
 
+    def test_ipython(self):
+        ureg = UnitRegistry()
+        x = 3.5 * ureg.Unit(UnitsContainer(meter=2, kilogram=1, second=-1))
+        self.assertEqual(x._repr_html_(),
+                         "3.5 kilogram meter<sup>2</sup>/second")
+        self.assertEqual(x._repr_latex_(),
+                         r'$3.5\ \frac{\mathrm{kilogram} \cdot '
+                         r'\mathrm{meter}^{2}}{\mathrm{second}}$')
+
     def test_to_base_units(self):
         x = self.Q_('1*inch')
         self.assertQuantityAlmostEqual(x.to_base_units(), self.Q_(0.0254, 'meter'))
diff --git a/pint/testsuite/test_unit.py b/pint/testsuite/test_unit.py
index 14555a8..599bb62 100644
--- a/pint/testsuite/test_unit.py
+++ b/pint/testsuite/test_unit.py
@@ -63,6 +63,13 @@ class TestUnit(QuantityTestCase):
             self.assertEqual('{0}'.format(x), result,
                              'Failed for {0}, {1}'.format(spec, result))
 
+    def test_ipython(self):
+        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}}$')
+
     def test_unit_mul(self):
         x = self.U_('m')
         self.assertEqual(x*1, self.Q_(1, 'm'))