Python atleast_1d Examples

Example #1

File: test_units.py Project: plant99/MetPy

def test_atleast1d_without_units():
    """Test that atleast_1d wrapper can handle plain arrays."""
    assert_array_equal(atleast_1d(1), np.array([1]))
    assert_array_equal(atleast_1d([
        1,
    ], [
        2,
    ]), np.array([[
        1,
    ], [
        2,
    ]]))

Example #2

File: test_units.py Project: dopplershift/MetPy

def test_atleast1d_without_units():
    """Test that atleast_1d wrapper can handle plain arrays."""
    assert_array_equal(atleast_1d(1), np.array([1]))