Merge pull request #11 from mattwthompson/fix-pint-1

Parse SimTK units with get_name instead of get_symbol

system/tests/test_utils.py

@@ -8,13 +8,23 @@
 
 u = pint.UnitRegistry()
 
-def test_simtk_to_pint():
+simtk_quantitites = [
+    4.0 * simtk_unit.nanometer,
+    5.0 * simtk_unit.angstrom,
+]
+pint_quantities = [
+    4.0 * u.nanometer,
+    5.0 * u.angstrom,
+]
+@pytest.mark.parametrize(
+    'simtk_quantity,pint_quantity',
+    [(s, p) for s, p in zip(simtk_quantitites, pint_quantities)]
+)
+def test_simtk_to_pint(simtk_quantity, pint_quantity):
     """Test conversion from SimTK Quantity to pint Quantity."""
-    simtk_quantity = 10.0 * simtk_unit.nanometer
-
-    pint_quantity = simtk_to_pint(simtk_quantity)
+    converted_pint_quantity = simtk_to_pint(simtk_quantity)
 
-    assert pint_quantity == 10.0 * u.nanometer
+    assert pint_quantity == converted_pint_quantity
 
 def test_pint_to_simtk():
     """Test conversion from pint Quantity to SimTK Quantity."""

system/utils.py

@@ -18,7 +18,7 @@ def simtk_to_pint(simtk_quantity):
     simtk_value = simtk_quantity.value_in_unit(simtk_unit)
 
     u = pint.UnitRegistry()
-    pint_unit = u(simtk_unit.get_symbol())
+    pint_unit = u(simtk_unit.get_name())
     pint_quantity = simtk_value * pint_unit
 
     return pint_quantity