26 lines
623 B
Python
26 lines
623 B
Python
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas import (
|
||
|
|
NA,
|
||
|
|
Series,
|
||
|
|
)
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("dtype", ["int64", "float64"])
|
||
|
|
def test_to_numpy_na_value(dtype):
|
||
|
|
# GH#48951
|
||
|
|
ser = Series([1, 2, NA, 4])
|
||
|
|
result = ser.to_numpy(dtype=dtype, na_value=0)
|
||
|
|
expected = np.array([1, 2, 0, 4], dtype=dtype)
|
||
|
|
tm.assert_numpy_array_equal(result, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_to_numpy_cast_before_setting_na():
|
||
|
|
# GH#50600
|
||
|
|
ser = Series([1])
|
||
|
|
result = ser.to_numpy(dtype=np.float64, na_value=np.nan)
|
||
|
|
expected = np.array([1.0])
|
||
|
|
tm.assert_numpy_array_equal(result, expected)
|