(self)
| 12 | |
| 13 | class TestToNumpy: |
| 14 | def test_to_numpy(self): |
| 15 | df = DataFrame({"A": [1, 2], "B": [3, 4.5]}) |
| 16 | expected = np.array([[1, 3], [2, 4.5]]) |
| 17 | result = df.to_numpy() |
| 18 | tm.assert_numpy_array_equal(result, expected) |
| 19 | |
| 20 | def test_to_numpy_dtype(self): |
| 21 | df = DataFrame({"A": [1, 2], "B": [3, 4.5]}) |