1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
| 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)
|
|
