update test

fbourgey · fbourgey · commit ed4db144efb8 · 2025-05-23T06:18:02.000-04:00
diff --git a/pandas/tests/frame/methods/test_to_numpy.py b/pandas/tests/frame/methods/test_to_numpy.py
@@ -50,9 +50,8 @@ def test_to_numpy_datetime_with_na(self):
         df = DataFrame(dti)
         df.iloc[0, 0] = NaT
         expected = np.array([[np.nan], [1.45169280e18], [1.45177920e18]])
-        assert np.allclose(
-            df.to_numpy(float, na_value=np.nan), expected, equal_nan=True
-        )
+        result = df.to_numpy(float, na_value=np.nan)
+        tm.assert_numpy_array_equal(result, expected)
 
         df = DataFrame(
             {
@@ -69,11 +68,12 @@ def test_to_numpy_datetime_with_na(self):
                 ],
             }
         )
-        arr = np.array(
+        expected = np.array(
             [
                 [0.00e00, 0.00e00, 1.00e00],
                 [8.64e04, np.nan, np.nan],
                 [np.nan, 8.64e04, 2.00e00],
             ]
         )
-        assert np.allclose(df.to_numpy(float, na_value=np.nan), arr, equal_nan=True)
+        result = df.to_numpy(float, na_value=np.nan)
+        tm.assert_numpy_array_equal(result, expected)

Original file line number	Diff line number	Diff line change
`@@ -50,9 +50,8 @@ def test_to_numpy_datetime_with_na(self):`
`50`	`50`	`df = DataFrame(dti)`
`51`	`51`	`df.iloc[0, 0] = NaT`
`52`	`52`	`expected = np.array([[np.nan], [1.45169280e18], [1.45177920e18]])`
`53`		`- assert np.allclose(`
`54`		`- df.to_numpy(float, na_value=np.nan), expected, equal_nan=True`
`55`		`- )`
	`53`	`+ result = df.to_numpy(float, na_value=np.nan)`
	`54`	`+ tm.assert_numpy_array_equal(result, expected)`
`56`	`55`
`57`	`56`	`df = DataFrame(`
`58`	`57`	`{`
`@@ -69,11 +68,12 @@ def test_to_numpy_datetime_with_na(self):`
`69`	`68`	`],`
`70`	`69`	`}`
`71`	`70`	`)`
`72`		`- arr = np.array(`
	`71`	`+ expected = np.array(`
`73`	`72`	`[`
`74`	`73`	`[0.00e00, 0.00e00, 1.00e00],`
`75`	`74`	`[8.64e04, np.nan, np.nan],`
`76`	`75`	`[np.nan, 8.64e04, 2.00e00],`
`77`	`76`	`]`
`78`	`77`	`)`
`79`		`- assert np.allclose(df.to_numpy(float, na_value=np.nan), arr, equal_nan=True)`
	`78`	`+ result = df.to_numpy(float, na_value=np.nan)`
	`79`	`+ tm.assert_numpy_array_equal(result, expected)`