def test_count(self): values = ['foo', 'foofoo', NA, 'foooofooofommmfoo'] result = strings.str_count(values, 'f[o]+') exp = [1, 2, NA, 4] tm.assert_almost_equal(result, exp) result = Series(values).str.count('f[o]+') tm.assert_isinstance(result, Series) tm.assert_almost_equal(result, exp) # mixed mixed = ['a', NA, 'b', True, datetime.today(), 'foo', None, 1, 2.] rs = strings.str_count(mixed, 'a') xp = [1, NA, 0, NA, NA, 0, NA, NA, NA] tm.assert_almost_equal(rs, xp) rs = Series(mixed).str.count('a') tm.assert_isinstance(rs, Series) tm.assert_almost_equal(rs, xp) # unicode values = [u('foo'), u('foofoo'), NA, u('foooofooofommmfoo')] result = strings.str_count(values, 'f[o]+') exp = [1, 2, NA, 4] tm.assert_almost_equal(result, exp) result = Series(values).str.count('f[o]+') tm.assert_isinstance(result, Series) tm.assert_almost_equal(result, exp)
def test_count(self): values = ["foo", "foofoo", NA, "foooofooofommmfoo"] result = strings.str_count(values, "f[o]+") exp = [1, 2, NA, 4] tm.assert_almost_equal(result, exp) result = Series(values).str.count("f[o]+") self.assert_(isinstance(result, Series)) tm.assert_almost_equal(result, exp) # mixed mixed = ["a", NA, "b", True, datetime.today(), "foo", None, 1, 2.0] rs = strings.str_count(mixed, "a") xp = [1, NA, 0, NA, NA, 0, NA, NA, NA] tm.assert_almost_equal(rs, xp) rs = Series(mixed).str.count("a") self.assert_(isinstance(rs, Series)) tm.assert_almost_equal(rs, xp) # unicode values = [u"foo", u"foofoo", NA, u"foooofooofommmfoo"] result = strings.str_count(values, "f[o]+") exp = [1, 2, NA, 4] tm.assert_almost_equal(result, exp) result = Series(values).str.count("f[o]+") self.assert_(isinstance(result, Series)) tm.assert_almost_equal(result, exp)