Skip to content

Commit b27f20a

Browse files
committed
update tests
1 parent e9d46c5 commit b27f20a

File tree

3 files changed

+6
-10
lines changed

3 files changed

+6
-10
lines changed

pandas/tests/io/excel/test_writers.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -858,10 +858,7 @@ def test_to_excel_multiindex_dates(self, merge_cells, path):
858858
unit = "us"
859859
if path.endswith(".ods"):
860860
unit = "s"
861-
new_index = [
862-
tsframe.index.as_unit(unit),
863-
np.arange(len(tsframe.index), dtype=np.int64),
864-
]
861+
new_index[0] = tsframe.index.as_unit(unit)
865862
tsframe.index = MultiIndex.from_arrays(new_index)
866863

867864
tsframe.index.names = ["time", "foo"]

pandas/tests/io/json/test_pandas.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -1630,9 +1630,8 @@ def test_read_timezone_information(self):
16301630
result = read_json(
16311631
StringIO('{"2019-01-01T11:00:00.000Z":88}'), typ="series", orient="index"
16321632
)
1633-
expected = Series(
1634-
[88], index=DatetimeIndex(["2019-01-01 11:00:00"], tz="UTC").as_unit("ns")
1635-
)
1633+
exp_dti = DatetimeIndex(["2019-01-01 11:00:00"], dtype="M8[ns, UTC]")
1634+
expected = Series([88], index=exp_dti)
16361635
tm.assert_series_equal(result, expected)
16371636

16381637
@pytest.mark.parametrize(

pandas/tests/reshape/test_pivot.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -437,10 +437,10 @@ def test_pivot_no_values(self):
437437
index=idx,
438438
)
439439
res = df.pivot_table(index=df.index.month, columns=Grouper(key="dt", freq="ME"))
440-
exp_columns = MultiIndex.from_tuples(
441-
[("A", pd.Timestamp("2011-01-31").as_unit("ns"))]
440+
exp_columns = MultiIndex.from_arrays(
441+
[["A"], pd.DatetimeIndex(["2011-01-31"], dtype="M8[ns]")],
442+
names=[None, "dt"],
442443
)
443-
exp_columns.names = [None, "dt"]
444444
exp = DataFrame(
445445
[3.25, 2.0], index=Index([1, 2], dtype=np.int32), columns=exp_columns
446446
)

0 commit comments

Comments
 (0)