Skip to content

Commit f8ac3e6

Browse files
committed
TST: fixed invalid DataFrame() initializations in tests
1 parent 9ef97f7 commit f8ac3e6

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

pandas/tests/frame/test_apply.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,8 @@ def test_apply_broadcast(self):
135135

136136
# scalars
137137
result = self.frame.apply(np.mean, result_type='broadcast')
138-
expected = DataFrame([self.frame.mean()], index=self.frame.index)
138+
expected = DataFrame([self.frame.mean()] * len(self.frame.index),
139+
index=self.frame.index)
139140
tm.assert_frame_equal(result, expected)
140141

141142
result = self.frame.apply(np.mean, axis=1, result_type='broadcast')

pandas/tests/indexes/test_multi.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -3215,8 +3215,7 @@ def test_nan_stays_float(self):
32153215
assert pd.isna(idx0.get_level_values(1)).all()
32163216
# the following failed in 0.14.1
32173217
assert pd.isna(idxm.get_level_values(1)[:-1]).all()
3218-
3219-
df0 = pd.DataFrame([[1, 2]], index=idx0)
3218+
df0 = pd.DataFrame([[1, 2]] * 2, index=idx0)
32203219
df1 = pd.DataFrame([[3, 4]], index=idx1)
32213220
dfm = df0 - df1
32223221
assert pd.isna(df0.index.get_level_values(1)).all()

pandas/tests/reshape/test_reshape.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -496,7 +496,7 @@ def test_preserve_categorical_dtype(self):
496496
cidx = pd.CategoricalIndex(list("xyz"), ordered=ordered)
497497
midx = pd.MultiIndex(levels=[['a'], cidx],
498498
labels=[[0, 0], [0, 1]])
499-
df = DataFrame([[10, 11]], index=midx)
499+
df = DataFrame([[10, 11]] * 2, index=midx)
500500

501501
expected = DataFrame([[1.0, 0.0, 0.0], [0.0, 1.0, 0.0]],
502502
index=midx, columns=cidx)

0 commit comments

Comments
 (0)