def test_timeseries_description_qs(self):
expected = self.parse('Q2052948 Dfi "eduskuntavaalit 1907"\n' +\
'Q1853901 Dfi "eduskuntavaalit 1908"\n' +\
'Q1571365 Dfi "eduskuntavaalit 1909"\n' +\
'Q1852888 Dfi "eduskuntavaalit 1910"\n' +\
'Q1571375 Dfi "eduskuntavaalit 1911"\n')
commands = [
'-d', '-t', '-p', 'eduskuntavaalit', 'fi', './tests/query.json'
]
outfile = StringIO()
wdlabelbuilder.process_arguments(commands, outfile)
self.assertMultiLineEqual(outfile.getvalue(), expected)
outfile.close()
def test_url_output(self):
expected = self.parse('https://tools.wmflabs.org/quickstatements/#v1=' +\
'Q2052948%09Len%09%22Finnish%20parliamentary%20election,' +\
'%201907%22%0AQ1853901%09Len%09%22Finnish%20parliamentary' +\
'%20election,%201908%22%0AQ1571365%09Len%09%22Finnish' +\
'%20parliamentary%20election,%201909%22%0AQ1852888%09Len' +\
'%09%22Finnish%20parliamentary%20election,%201910%22' +\
'%0AQ1571375%09Len%09%22Finnish%20parliamentary%20election,%201911%22\n')
commands = ['-u', '-l', 'en', './tests/query.json']
outfile = StringIO()
wdlabelbuilder.process_arguments(commands, outfile)
self.assertMultiLineEqual(outfile.getvalue(), expected)
outfile.close()
def test_timeseries_label_prefix_json(self):
expected = self.parse(
'[{"lang": "fi","item": "Q2052948","itemLabel": "eduskuntavaalit 1907"},'+\
'{"lang": "fi","item": "Q1853901","itemLabel": "eduskuntavaalit 1908"},'+\
'{"lang": "fi","item": "Q1571365","itemLabel": "eduskuntavaalit 1909"},'+\
'{"lang": "fi","item": "Q1852888","itemLabel": "eduskuntavaalit 1910"},'+\
'{"lang": "fi","item": "Q1571375","itemLabel": "eduskuntavaalit 1911"}]')
commands = [
'-l', '-t', '-j', '-p', 'eduskuntavaalit', 'fi',
'./tests/output.json'
]
outfile = StringIO()
wdlabelbuilder.process_arguments(commands, outfile)
self.assertMultiLineEqual(outfile.getvalue(), expected)
outfile.close()
def test_timeseries_alias_qs_to_file(self):
expected = self.parse('Q2052948 Afi "eduskuntavaalit 1907"\n' +\
'Q1853901 Afi "eduskuntavaalit 1908"\n' +\
'Q1571365 Afi "eduskuntavaalit 1909"\n' +\
'Q1852888 Afi "eduskuntavaalit 1910"\n' +\
'Q1571375 Afi "eduskuntavaalit 1911"\n')
commands = [
'-a', '-t', '-p', 'eduskuntavaalit', '-o', './tests/output', 'fi',
'./tests/query.json'
]
wdlabelbuilder.process_arguments(commands)
with open('./tests/output', 'r') as infile:
data = infile.read()
self.assertMultiLineEqual(data, expected)
os.remove('./tests/output')
def test_timeseries_label_suffix_json(self):
expected = [
{"lang": "fi","item": "Q2052948","itemLabel": "1907 eduskuntavaalit"},\
{"lang": "fi","item": "Q1853901","itemLabel": "1908 eduskuntavaalit"},\
{"lang": "fi","item": "Q1571365","itemLabel": "1909 eduskuntavaalit"},\
{"lang": "fi","item": "Q1852888","itemLabel": "1910 eduskuntavaalit"},\
{"lang": "fi","item": "Q1571375","itemLabel": "1911 eduskuntavaalit"}]
commands = [
'-l', '-t', '-j', '-s', 'eduskuntavaalit', '-o',
'./tests/test_output.json', 'fi', './tests/query.json'
]
wdlabelbuilder.process_arguments(commands)
with open('./tests/test_output.json', 'r') as infile:
data = json.load(infile)
self.assertListEqual(data, expected)
os.remove('./tests/test_output.json')
def test_no_lda(self):
''' Input arguments without -l, -d or -a, expect SystemExit. '''
commands = ['fi', './tests/query.json']
with self.assertRaises(SystemExit):
args = wdlabelbuilder.process_arguments(commands)
def test_wrong_input_filepath(self):
''' Input a wrong input filepath, expect SystemExit. '''
commands = ['-l', 'fi', './tests/woefichtr.json']
with self.assertRaises(SystemExit):
args = wdlabelbuilder.process_arguments(commands)
def test_non_json_input_file(self):
''' Input a non-json file, expect SystemExit. '''
commands = ['-l', 'fi', './tests/non_json_file']
with self.assertRaises(SystemExit):
args = wdlabelbuilder.process_arguments(commands)
def test_wrong_input_json_titles(self):
''' Input an empty json file, expect SystemExit. '''
commands = ['-l', 'fi', '-q', 'asd', '-n', 'asd', './tests/query.json']
with self.assertRaises(SystemExit):
args = wdlabelbuilder.process_arguments(commands)