def get_query(self): query = self.entry.get_text() return capitalize_query(query)
def test_capitalize_string(self): self.assertEquals(u"Ddt", capitalize_query("ddt")) self.assertEquals(u"DDT", capitalize_query("DDT")) self.assertEquals(u"DDT Music", capitalize_query("DDT music"))
def test_capitalize_url(self): self.assertEquals(u"http://Madonna", capitalize_query("http://Madonna"))
def test_capitalize_None(self): self.assertEquals(None, capitalize_query(None)) self.assertEquals("", capitalize_query(""))