def test_atleast2d_with_units(): """Test that atleast_2d wrapper can handle plain array with units.""" assert_array_equal(atleast_2d(1 * units.degC), np.array([[1]]) * units.degC)
def test_atleast2d_without_units(): """Test that atleast_2d wrapper can handle plain arrays.""" assert_array_equal(atleast_2d(1), np.array([[1]]))
def test_atleast2d_with_units(): """Test that atleast_2d wrapper can handle plain array with units.""" assert_array_equal( atleast_2d(1 * units.degC), np.array([[1]]) * units.degC)