def test_country_code(self): a = Babe().pull(string=self.s, format='csv') a = a.geoip_country_code() self.assertEquals(a.to_string(), self.s2)
def test_country_code(self): a = Babe().pull(stream=StringIO(self.s), format='csv') a = a.geoip_country_code() buf = StringIO() a.push(stream=buf, format="csv") self.assertEquals(buf.getvalue(), self.s2)