def test_image_list(self): splitter, joiner = headline._create_splitter(u"br") klass = HEADLINE_CLASSES["class_name"] images = headline._image_list("Test test", klass, splitter, object=False) # Image creating and html representation i = 0 for image in images: i += 1 self.assertEqual( image, """<img alt="Test test" src="./headline-91cafd03cea08d1cd558f849e530da18.png" />""", "(Template want work or hash algorythm changed)", ) self.assertEqual(i, 1) splitter, joiner = headline._create_splitter(u"all") images = headline._image_list("test test", klass, splitter, object=True) # Image creating and dict representation for image in images: i += 1 self.assert_(path.isfile(image["url"]), "File is not created") self.assert_(image["width"], "There is no returned width") self.assert_(image["height"], "There is no returned height") self.assert_(image["text"], "There is no returned text") self.assertEqual(image["text"], "test", "Returned text is not normal") real_image = Image.open(image["url"]) size = real_image.size self.assertEqual( Image.open(image["url"]).size, (image["width"], image["height"]), "Real image size is not normal" ) colors = real_image.getcolors() color_count = len(colors) self.assert_(8 < color_count < 150, "Color count on image is not normal normal is 8..64") self.assertEqual(i, 3)
def test_create_splitter(self): splitter, joiner = headline._create_splitter(u"br") # Br splitter self.assertEqual(splitter.split("<br /><br/><br><br />"), ["", "", "", "", ""], "Br splitter failed") self.assertEqual( splitter.split("br br br<br />br br<br> br br"), ["br br br", "br br", " br br"], "Br splitter failed" ) self.assertEqual(splitter.split("br"), ["br"], "Br splitter failed: single") self.assertEqual(joiner.join(("", "")), "<br />", "Br joiner failed") self.assertEqual(joiner.join(("",)), "", "Br joiner failed: single") # All splitter splitter, joiner = headline._create_splitter(u"all") self.assertEqual(splitter.split("test test<br />test"), ["test", "test", "test"], "All splitter failed: ") self.assertEqual(splitter.split("test test<br /> test"), ["test", "test", "", "test"], "All splitter failed: ") self.assertEqual(splitter.split("testtesttest"), ["testtesttest"], "All splitter failed: single") # None splitter self.assertEqual(headline._create_splitter(u"none"), (False, ""), "None splitter failed: none") self.assertEqual(headline._create_splitter(u"other"), (False, ""), "None splitter failed: other") self.assertEqual(headline._create_splitter(u""), (False, ""), "None splitter failed: ''")