def test_format_empty_value_parameter_with_empty(self): expected = u'|parameter= \n' param = 'parameter' self.assertEqual(ImageInfo.format_empty_value_parameter(param, ''), expected) self.assertEqual(ImageInfo.format_empty_value_parameter(param, []), expected) self.assertEqual(ImageInfo.format_empty_value_parameter(param, None), expected)
def test_format_multi_value_parameter_with_empty(self): expected = u'|parameter= \n' param = 'parameter' self.assertEqual(ImageInfo.format_multi_value_parameter(param, ''), expected) self.assertEqual(ImageInfo.format_multi_value_parameter(param, []), expected) self.assertEqual(ImageInfo.format_multi_value_parameter(param, None), expected)
def test_format_depicted_with_single_and_inv_nr(self): expected = u'{{depicted person|p1|style=plain text|comment=invNr}}\n' value = [ 'p1', ] self.assertEqual(ImageInfo.format_depicted(value, inv_nr='invNr'), expected)
class TestFormatCategory(unittest.TestCase): """Test ImageInfo.format_category()""" def setUp(self): self.caption = u'A caption' self.image_info = ImageInfo() self.image_info.categories = [] def assert_same_category_content(self, actual, expected): """Assert that a category block is same, up to order of its entries.""" actual_lines = actual.split('\n') expected_lines = expected.split('\n') # check first line self.assertEqual(actual_lines[0], expected_lines[0]) # check contents of rest irrespective of order self.assertItemsEqual(actual_lines[1:], expected_lines[1:]) def test_format_category_empty(self): expected_text = '' expected_printed = [] self.assertEqual( self.image_info.format_category(self.caption, []), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_single(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat]]\n' expected_printed = ['Cat'] categories = ['Cat'] self.assertEqual( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_duplicate(self): """Ensure internal duplicates are not outputted.""" expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat]]\n' expected_printed = ['Cat'] categories = ['Cat', 'Cat'] self.assertEqual( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_reprint(self): """Ensure already outputted categories are not re-outputted.""" expected_text = '' expected_printed = ['Cat'] self.image_info.categories = ['Cat'] categories = ['Cat', ] self.assertEqual( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_multiple(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat1]]\n' \ u'[[Category:Cat2]]\n' \ u'[[Category:Cat3]]\n' expected_printed = ['Cat1', 'Cat2', 'Cat3'] categories = ['Cat1', 'Cat2', 'Cat3'] self.assert_same_category_content( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_with_prefix(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Prefix-Cat1]]\n' \ u'[[Category:Prefix-Cat2]]\n' expected_printed = ['Cat1', 'Cat2'] categories = ['Cat1', 'Cat2'] prefix = u'Prefix-' self.assert_same_category_content( self.image_info.format_category(self.caption, categories, prefix=prefix), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_no_de_duplication(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat1]]\n' \ u'[[Category:Cat2]]\n' expected_printed = [] categories = ['Cat1', 'Cat1', 'Cat2'] self.assert_same_category_content( self.image_info.format_category(self.caption, categories, de_duplicate=False), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed)
def test_format_depicted_with_many(self): expected = u'{{depicted person|p1|p2|p3|p4|p5|p6|p7|p8|p9|style=plain text}}\n' \ u'{{depicted person|p10|style=plain text}}\n' value = ['p1', 'p2', 'p3', 'p4', 'p5', 'p6', 'p7', 'p8', 'p9', 'p10'] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_depicted_with_single(self): expected = u'{{depicted person|p1|style=plain text}}\n' value = [ 'p1', ] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_multi_value_parameter_with_multiple_and_creator(self): expected = u'|parameter= * val1\n{{Creator:val2}}\n* val3\n' value = ['val1', '{{Creator:val2}}', 'val3'] self.assertEqual( ImageInfo.format_multi_value_parameter('parameter', value), expected)
def test_format_multi_value_parameter_with_single(self): expected = u'|parameter= value\n' self.assertEqual( ImageInfo.format_multi_value_parameter('parameter', [ 'value', ]), expected)
class TestProcessCategories(unittest.TestCase): """Test ImageInfo.process_categories().""" def setUp(self): self.image_info = ImageInfo() self.image_info.meta_categories = [] self.image_info.obj_data = dict.fromkeys( [u'cat_event', u'cat_artist', u'cat_depicted', u'cat_obj']) def runAsserts(self, expected_output, expected_cats, expected_meta_cats): """Hide complicated paths needed for asserts.""" expected_output = '\n%s\n' % expected_output # make input easier self.assertEqual(self.image_info.categories_as_text, expected_output) self.assertItemsEqual(self.image_info.categories, expected_cats) self.assertItemsEqual(self.image_info.meta_categories, expected_meta_cats) def test_process_categories_only_meta_adds_meta(self): """Only meta triggers "without cats", and is not added to cats.""" expected_output = \ '<!--Maintanance categories-->\n'\ '[[Category:Media contributed by LSH: without any categories]]\n'\ '[[Category:Media contributed by LSH: meta_1]]' expected_cats = [] expected_meta_cats = ['meta_1', 'without any categories'] self.image_info.meta_categories = ['meta_1'] self.image_info.process_categories(None, None), self.runAsserts(expected_output, expected_cats, expected_meta_cats) def test_process_categories_only_photographer_adds_meta(self): """Only photographer triggers "without cats", but is added to cats.""" expected_output = \ '<!--Photographer category-->\n' \ '[[Category:cat_photographer]]\n\n' \ '<!--Maintanance categories-->\n'\ '[[Category:Media contributed by LSH: without any categories]]' expected_cats = ['cat_photographer'] expected_meta_cats = ['without any categories'] self.image_info.process_categories(None, 'cat_photographer'), self.runAsserts(expected_output, expected_cats, expected_meta_cats) def test_process_categories_stich_cats(self): """Stich does not triggers "without cats", but is added to cats.""" expected_output = \ '<!--Photograph categories-->\n'\ '[[Category:stich_1]]\n'\ '[[Category:stich_2]]' expected_cats = ['stich_1', 'stich_2'] expected_meta_cats = [] self.image_info.process_categories(['stich_1', 'stich_2'], None), self.runAsserts(expected_output, expected_cats, expected_meta_cats) def test_process_categories_all_obj_data(self): """Test full obj_data category structure.""" expected_output = \ '<!--Event categories-->\n'\ '[[Category:event_1]]\n\n'\ '<!--Artist categories-->\n'\ '[[Category:artist_1]]\n\n'\ '<!--Depicted categories-->\n'\ '[[Category:depicted_1]]\n\n'\ '<!--Object categories-->\n'\ '[[Category:obj_2]]\n'\ '[[Category:obj_1]]' expected_cats = ['event_1', 'artist_1', 'depicted_1', 'obj_1', 'obj_2'] expected_meta_cats = [] self.image_info.obj_data = { u'cat_event': ['event_1'], u'cat_artist': ['artist_1'], u'cat_depicted': ['depicted_1'], u'cat_obj': ['obj_1', 'obj_2'] } self.image_info.process_categories(None, None), self.runAsserts(expected_output, expected_cats, expected_meta_cats)
def test_format_multi_value_parameter_with_multiple(self): expected = u'|parameter= * val1\n* val2\n* val3\n' value = ['val1', 'val2', 'val3'] self.assertEqual( ImageInfo.format_multi_value_parameter('parameter', value), expected)
def test_format_multi_value_parameter_with_single(self): expected = u'|parameter= value\n' self.assertEqual( ImageInfo.format_multi_value_parameter('parameter', ['value', ]), expected)
def test_format_empty_value_parameter_with_value(self): expected = u'|parameter= value\n' self.assertEqual( ImageInfo.format_empty_value_parameter('parameter', 'value'), expected)
def setUp(self): self.image_info = ImageInfo() self.image_info.meta_categories = [] self.image_info.obj_data = dict.fromkeys( [u'cat_event', u'cat_artist', u'cat_depicted', u'cat_obj'])
class TestProcessCategories(unittest.TestCase): """Test ImageInfo.process_categories().""" def setUp(self): self.image_info = ImageInfo() self.image_info.meta_categories = [] self.image_info.obj_data = dict.fromkeys( [u'cat_event', u'cat_artist', u'cat_depicted', u'cat_obj']) def runAsserts(self, expected_output, expected_cats, expected_meta_cats): """Hide complicated paths needed for asserts.""" expected_output = '\n%s\n' % expected_output # make input easier self.assertEqual(self.image_info.categories_as_text, expected_output) self.assertItemsEqual(self.image_info.categories, expected_cats) self.assertItemsEqual(self.image_info.meta_categories, expected_meta_cats) def test_process_categories_only_meta_adds_meta(self): """Only meta triggers "without cats", and is not added to cats.""" expected_output = \ '<!--Maintanance categories-->\n'\ '[[Category:Media contributed by LSH: without any categories]]\n'\ '[[Category:Media contributed by LSH: meta_1]]' expected_cats = [] expected_meta_cats = ['meta_1', 'without any categories'] self.image_info.meta_categories = ['meta_1'] self.image_info.process_categories(None, None), self.runAsserts(expected_output, expected_cats, expected_meta_cats) def test_process_categories_only_photographer_adds_meta(self): """Only photographer triggers "without cats", but is added to cats.""" expected_output = \ '<!--Photographer category-->\n' \ '[[Category:cat_photographer]]\n\n' \ '<!--Maintanance categories-->\n'\ '[[Category:Media contributed by LSH: without any categories]]' expected_cats = ['cat_photographer'] expected_meta_cats = ['without any categories'] self.image_info.process_categories(None, 'cat_photographer'), self.runAsserts(expected_output, expected_cats, expected_meta_cats) def test_process_categories_stich_cats(self): """Stich does not triggers "without cats", but is added to cats.""" expected_output = \ '<!--Photograph categories-->\n'\ '[[Category:stich_1]]\n'\ '[[Category:stich_2]]' expected_cats = ['stich_1', 'stich_2'] expected_meta_cats = [] self.image_info.process_categories(['stich_1', 'stich_2'], None), self.runAsserts(expected_output, expected_cats, expected_meta_cats) def test_process_categories_all_obj_data(self): """Test full obj_data category structure.""" expected_output = \ '<!--Event categories-->\n'\ '[[Category:event_1]]\n\n'\ '<!--Artist categories-->\n'\ '[[Category:artist_1]]\n\n'\ '<!--Depicted categories-->\n'\ '[[Category:depicted_1]]\n\n'\ '<!--Object categories-->\n'\ '[[Category:obj_2]]\n'\ '[[Category:obj_1]]' expected_cats = ['event_1', 'artist_1', 'depicted_1', 'obj_1', 'obj_2'] expected_meta_cats = [] self.image_info.obj_data = { u'cat_event': ['event_1'], u'cat_artist': ['artist_1'], u'cat_depicted': ['depicted_1'], u'cat_obj': ['obj_1', 'obj_2'] } self.image_info.process_categories(None, None), self.runAsserts(expected_output, expected_cats, expected_meta_cats)
def setUp(self): self.caption = u'A caption' self.image_info = ImageInfo() self.image_info.categories = []
def setUp(self): self.caption = u'A caption' self.image_info = ImageInfo() self.image_info.categories = []
def test_format_empty_value_parameter_with_value(self): expected = u'|parameter= value\n' self.assertEqual( ImageInfo.format_empty_value_parameter('parameter', 'value'), expected)
def test_format_multi_value_parameter_with_multiple_and_creator(self): expected = u'|parameter= * val1\n{{Creator:val2}}\n* val3\n' value = ['val1', '{{Creator:val2}}', 'val3'] self.assertEqual( ImageInfo.format_multi_value_parameter('parameter', value), expected)
def setUp(self): self.image_info = ImageInfo() self.image_info.meta_categories = [] self.image_info.obj_data = dict.fromkeys( [u'cat_event', u'cat_artist', u'cat_depicted', u'cat_obj'])
def test_format_depicted_with_empty(self): expected = u'' value = [] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_multi_value_parameter_with_multiple(self): expected = u'|parameter= * val1\n* val2\n* val3\n' value = ['val1', 'val2', 'val3'] self.assertEqual( ImageInfo.format_multi_value_parameter('parameter', value), expected)
def test_format_depicted_with_single(self): expected = u'{{depicted person|p1|style=plain text}}\n' value = ['p1', ] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_depicted_with_empty(self): expected = u'' value = [] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_depicted_with_few(self): expected = u'{{depicted person|p1|p2|p3|style=plain text}}\n' value = ['p1', 'p2', 'p3'] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_depicted_with_few(self): expected = u'{{depicted person|p1|p2|p3|style=plain text}}\n' value = ['p1', 'p2', 'p3'] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_depicted_with_many(self): expected = u'{{depicted person|p1|p2|p3|p4|p5|p6|p7|p8|p9|style=plain text}}\n' \ u'{{depicted person|p10|style=plain text}}\n' value = ['p1', 'p2', 'p3', 'p4', 'p5', 'p6', 'p7', 'p8', 'p9', 'p10'] self.assertEqual(ImageInfo.format_depicted(value), expected)
def test_format_depicted_with_single_and_inv_nr(self): expected = u'{{depicted person|p1|style=plain text|comment=invNr}}\n' value = ['p1', ] self.assertEqual(ImageInfo.format_depicted(value, inv_nr='invNr'), expected)
class TestFormatCategory(unittest.TestCase): """Test ImageInfo.format_category()""" def setUp(self): self.caption = u'A caption' self.image_info = ImageInfo() self.image_info.categories = [] def assert_same_category_content(self, actual, expected): """Assert that a category block is same, up to order of its entries.""" actual_lines = actual.split('\n') expected_lines = expected.split('\n') # check first line self.assertEqual(actual_lines[0], expected_lines[0]) # check contents of rest irrespective of order self.assertItemsEqual(actual_lines[1:], expected_lines[1:]) def test_format_category_empty(self): expected_text = '' expected_printed = [] self.assertEqual(self.image_info.format_category(self.caption, []), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_single(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat]]\n' expected_printed = ['Cat'] categories = ['Cat'] self.assertEqual( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_duplicate(self): """Ensure internal duplicates are not outputted.""" expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat]]\n' expected_printed = ['Cat'] categories = ['Cat', 'Cat'] self.assertEqual( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_reprint(self): """Ensure already outputted categories are not re-outputted.""" expected_text = '' expected_printed = ['Cat'] self.image_info.categories = ['Cat'] categories = [ 'Cat', ] self.assertEqual( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_multiple(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat1]]\n' \ u'[[Category:Cat2]]\n' \ u'[[Category:Cat3]]\n' expected_printed = ['Cat1', 'Cat2', 'Cat3'] categories = ['Cat1', 'Cat2', 'Cat3'] self.assert_same_category_content( self.image_info.format_category(self.caption, categories), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_with_prefix(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Prefix-Cat1]]\n' \ u'[[Category:Prefix-Cat2]]\n' expected_printed = ['Cat1', 'Cat2'] categories = ['Cat1', 'Cat2'] prefix = u'Prefix-' self.assert_same_category_content( self.image_info.format_category(self.caption, categories, prefix=prefix), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed) def test_format_category_no_de_duplication(self): expected_text = u'\n<!--A caption-->\n' \ u'[[Category:Cat1]]\n' \ u'[[Category:Cat2]]\n' expected_printed = [] categories = ['Cat1', 'Cat1', 'Cat2'] self.assert_same_category_content( self.image_info.format_category(self.caption, categories, de_duplicate=False), expected_text) self.assertItemsEqual(self.image_info.categories, expected_printed)
def test_format_depicted_with_empty_and_inv_nr(self): expected = u'' value = [] self.assertEqual(ImageInfo.format_depicted(value, inv_nr='invNr'), expected)
def test_format_depicted_with_empty_and_inv_nr(self): expected = u'' value = [] self.assertEqual(ImageInfo.format_depicted(value, inv_nr='invNr'), expected)