Skip to content

Commit aa9d9b9

Browse files
committed
Merge pull request #5139 from jreback/fillna_string
TST: fillna(values) equiv of replace(np.nan,values)
2 parents 42f42c4 + 2097dc2 commit aa9d9b9

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

pandas/tests/test_frame.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3744,31 +3744,31 @@ def tuple_generator(length):
37443744
for i in range(length):
37453745
letters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
37463746
yield (i, letters[i % len(letters)], i/length)
3747-
3747+
37483748
columns_names = ['Integer', 'String', 'Float']
37493749
columns = [[i[j] for i in tuple_generator(10)] for j in range(len(columns_names))]
37503750
data = {'Integer': columns[0], 'String': columns[1], 'Float': columns[2]}
37513751
expected = DataFrame(data, columns=columns_names)
3752-
3752+
37533753
generator = tuple_generator(10)
37543754
result = DataFrame.from_records(generator, columns=columns_names)
37553755
assert_frame_equal(result, expected)
3756-
3756+
37573757
def test_from_records_lists_generator(self):
37583758
def list_generator(length):
37593759
for i in range(length):
37603760
letters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
37613761
yield [i, letters[i % len(letters)], i/length]
3762-
3762+
37633763
columns_names = ['Integer', 'String', 'Float']
37643764
columns = [[i[j] for i in list_generator(10)] for j in range(len(columns_names))]
37653765
data = {'Integer': columns[0], 'String': columns[1], 'Float': columns[2]}
37663766
expected = DataFrame(data, columns=columns_names)
3767-
3767+
37683768
generator = list_generator(10)
37693769
result = DataFrame.from_records(generator, columns=columns_names)
37703770
assert_frame_equal(result, expected)
3771-
3771+
37723772
def test_from_records_columns_not_modified(self):
37733773
tuples = [(1, 2, 3),
37743774
(1, 2, 3),
@@ -6745,6 +6745,13 @@ def test_fillna_dtype_conversion(self):
67456745
expected = DataFrame('nan',index=lrange(3),columns=['A','B'])
67466746
assert_frame_equal(result, expected)
67476747

6748+
# equiv of replace
6749+
df = DataFrame(dict(A = [1,np.nan], B = [1.,2.]))
6750+
for v in ['',1,np.nan,1.0]:
6751+
expected = df.replace(np.nan,v)
6752+
result = df.fillna(v)
6753+
assert_frame_equal(result, expected)
6754+
67486755
def test_ffill(self):
67496756
self.tsframe['A'][:5] = nan
67506757
self.tsframe['A'][-5:] = nan

0 commit comments

Comments
 (0)