| 421 | tm.assert_frame_equal(written_and_read_again.set_index("index"), expected) |
| 422 | |
| 423 | def test_write_dta6(self, datapath, temp_file): |
| 424 | original = self.read_csv(datapath("io", "data", "stata", "stata3.csv")) |
| 425 | original.index.name = "index" |
| 426 | original.index = original.index.astype(np.int32) |
| 427 | original["year"] = original["year"].astype(np.int32) |
| 428 | original["quarter"] = original["quarter"].astype(np.int32) |
| 429 | |
| 430 | path = temp_file |
| 431 | original.to_stata(path, convert_dates=None) |
| 432 | written_and_read_again = self.read_dta(path) |
| 433 | tm.assert_frame_equal( |
| 434 | written_and_read_again.set_index("index"), |
| 435 | original, |
| 436 | check_index_type=False, |
| 437 | ) |
| 438 | |
| 439 | @pytest.mark.parametrize("version", [114, 117, 118, 119, None]) |
| 440 | def test_read_write_dta10(self, version, temp_file, using_infer_string): |