def test_to_value(): a = YTArray([1.0, 2.0, 3.0], "kpc") assert_equal(a.to_value(), np.array([1.0, 2.0, 3.0])) assert_equal(a.to_value(), a.value) assert_equal(a.to_value("km"), a.in_units("km").value) b = YTQuantity(5.5, "Msun") assert_equal(b.to_value(), 5.5) assert_equal(b.to_value("g"), b.in_units("g").value)
def test_display_ytarray(): arr = YTArray([1,2,3], 'cm') widget = display_ytarray(arr) dropdown = widget.children[-1] dropdown.value = 'm' # Check that our original array did *not* change assert_equal(arr.to_value(), np.array([1.0, 2.0, 3.0])) # Check our values did change assert_equal([float(_.value) for _ in widget.children[:-1]], arr.to_value()/100.0)