def test_cat(self):
one = ['a', 'a', 'b', 'b', 'c', NA]
two = ['a', NA, 'b', 'd', 'foo', NA]
# single array
result = strings.str_cat(one)
self.assert_(isnull(result))
result = strings.str_cat(one, na_rep='NA')
exp = 'aabbcNA'
self.assertEquals(result, exp)
result = strings.str_cat(one, na_rep='-')
exp = 'aabbc-'
self.assertEquals(result, exp)
result = strings.str_cat(one, sep='_', na_rep='NA')
exp = 'a_a_b_b_c_NA'
self.assertEquals(result, exp)
# Multiple arrays
result = strings.str_cat(one, [two], na_rep='NA')
exp = ['aa', 'aNA', 'bb', 'bd', 'cfoo', 'NANA']
self.assert_(np.array_equal(result, exp))
result = strings.str_cat(one, two)
exp = ['aa', NA, 'bb', 'bd', 'cfoo', NA]
tm.assert_almost_equal(result, exp)
def test_cat(self):
one = ["a", "a", "b", "b", "c", NA]
two = ["a", NA, "b", "d", "foo", NA]
# single array
result = strings.str_cat(one)
self.assert_(isnull(result))
result = strings.str_cat(one, na_rep="NA")
exp = "aabbcNA"
self.assertEquals(result, exp)
result = strings.str_cat(one, na_rep="-")
exp = "aabbc-"
self.assertEquals(result, exp)
result = strings.str_cat(one, sep="_", na_rep="NA")
exp = "a_a_b_b_c_NA"
self.assertEquals(result, exp)
# Multiple arrays
result = strings.str_cat(one, [two], na_rep="NA")
exp = ["aa", "aNA", "bb", "bd", "cfoo", "NANA"]
self.assert_(np.array_equal(result, exp))
result = strings.str_cat(one, two)
exp = ["aa", NA, "bb", "bd", "cfoo", NA]
tm.assert_almost_equal(result, exp)
