1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| from io import StringIO
|
| import numpy as np
| import pytest
|
| import pandas as pd
| from pandas.tests.extension.base.base import BaseExtensionTests
|
|
| class BaseParsingTests(BaseExtensionTests):
| @pytest.mark.parametrize("engine", ["c", "python"])
| def test_EA_types(self, engine, data):
| df = pd.DataFrame({"with_dtype": pd.Series(data, dtype=str(data.dtype))})
| csv_output = df.to_csv(index=False, na_rep=np.nan)
| result = pd.read_csv(
| StringIO(csv_output), dtype={"with_dtype": str(data.dtype)}, engine=engine
| )
| expected = df
| self.assert_frame_equal(result, expected)
|
|