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)
Пример #2
0
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)