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)