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)
