def test_pint():
from pint import UnitRegistry
ureg = UnitRegistry()
p_arr = np.arange(10) * ureg.km / ureg.hr
yt_arr = YTArray(np.arange(10), "km/hr")
yt_arr2 = YTArray.from_pint(p_arr)
p_quan = 10. * ureg.g**0.5 / (ureg.mm**3)
yt_quan = YTQuantity(10., "sqrt(g)/mm**3")
yt_quan2 = YTQuantity.from_pint(p_quan)
assert_array_equal(p_arr, yt_arr.to_pint())
assert_equal(p_quan, yt_quan.to_pint())
assert_array_equal(yt_arr, YTArray.from_pint(p_arr))
assert_array_equal(yt_arr, yt_arr2)
assert_equal(p_quan.magnitude, yt_quan.to_pint().magnitude)
assert_equal(p_quan, yt_quan.to_pint())
assert_equal(yt_quan, YTQuantity.from_pint(p_quan))
assert_equal(yt_quan, yt_quan2)
assert_array_equal(yt_arr, YTArray.from_pint(yt_arr.to_pint()))
assert_equal(yt_quan, YTQuantity.from_pint(yt_quan.to_pint()))
