Skip to content

Commit 414010b

Browse files
author
MarcoGorelli
committed
avoid some upcasting when its not the purpose of the test
1 parent 502919e commit 414010b

File tree

4 files changed

+10
-6
lines changed

4 files changed

+10
-6
lines changed

pandas/tests/frame/test_query_eval.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -448,7 +448,7 @@ def test_date_index_query(self):
448448
def test_date_index_query_with_NaT(self):
449449
engine, parser = self.engine, self.parser
450450
n = 10
451-
df = DataFrame(np.random.randn(n, 3))
451+
df = DataFrame(np.random.randn(n, 3)).astype({0: object})
452452
df["dates1"] = date_range("1/1/2012", periods=n)
453453
df["dates3"] = date_range("1/1/2014", periods=n)
454454
df.iloc[0, 0] = pd.NaT
@@ -809,7 +809,7 @@ def test_date_index_query_with_NaT(self):
809809
engine, parser = self.engine, self.parser
810810
n = 10
811811
df = DataFrame(np.random.randn(n, 3))
812-
df["dates1"] = date_range("1/1/2012", periods=n)
812+
df["dates1"] = date_range("1/1/2012", periods=n).astype({0: object})
813813
df["dates3"] = date_range("1/1/2014", periods=n)
814814
df.iloc[0, 0] = pd.NaT
815815
return_value = df.set_index("dates1", inplace=True, drop=True)

pandas/tests/frame/test_reductions.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -448,11 +448,15 @@ def test_var_std(self, datetime_frame):
448448
@pytest.mark.parametrize("meth", ["sem", "var", "std"])
449449
def test_numeric_only_flag(self, meth):
450450
# GH 9201
451-
df1 = DataFrame(np.random.randn(5, 3), columns=["foo", "bar", "baz"])
451+
df1 = DataFrame(np.random.randn(5, 3), columns=["foo", "bar", "baz"]).astype(
452+
{"foo": object}
453+
)
452454
# set one entry to a number in str format
453455
df1.loc[0, "foo"] = "100"
454456

455-
df2 = DataFrame(np.random.randn(5, 3), columns=["foo", "bar", "baz"])
457+
df2 = DataFrame(np.random.randn(5, 3), columns=["foo", "bar", "baz"]).astype(
458+
{"foo": object}
459+
)
456460
# set one entry to a non-number str
457461
df2.loc[0, "foo"] = "a"
458462

pandas/tests/groupby/test_timegrouper.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def test_groupby_with_timegrouper(self):
102102
index=date_range(
103103
"20130901", "20131205", freq="5D", name="Date", inclusive="left"
104104
),
105-
)
105+
).astype({"Buyer": object})
106106
expected.iloc[0, 0] = "CarlCarlCarl"
107107
expected.iloc[6, 0] = "CarlCarl"
108108
expected.iloc[18, 0] = "Joe"

pandas/tests/series/methods/test_replace.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def test_replace_explicit_none(self):
1616
expected = pd.Series([0, 0, None], dtype=object)
1717
tm.assert_series_equal(result, expected)
1818

19-
df = pd.DataFrame(np.zeros((3, 3)))
19+
df = pd.DataFrame(np.zeros((3, 3))).astype({2: object})
2020
df.iloc[2, 2] = ""
2121
result = df.replace("", None)
2222
expected = pd.DataFrame(

0 commit comments

Comments
 (0)