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)