def test_output_json(random1, random2): """Test to make sure the JSON is written out correctly""" data = [{ 'link': 'courage', 'zelda': 'wisdom', 'ganon': 'power', 'num': 3 }, { 'random_one': random1, 'random_two': random2 }] output_file = WriteTester() script_output.write_json(output_file, data) lines = output_file.lines() assert len(lines) == 5 assert json.loads(''.join(lines)) assert lines[0] == '[\n' assert lines[2] == ',\n' assert lines[4] == '\n]\n' dict1 = json.loads(lines[1]) dict2 = json.loads(lines[3]) assert dict1['link'] == 'courage' assert dict1['zelda'] == 'wisdom' assert dict1['ganon'] == 'power' assert dict1['num'] == 3 assert dict2['random_one'] == random1 assert dict2['random_two'] == random2
def test_output_no_json(): """Test writing empty JSON""" data = {} output_file = WriteTester() script_output.write_json(output_file, data) lines = output_file.lines() assert len(lines) == 2 assert lines[0] == '[\n' assert lines[1] == '\n]\n'
def test_output_no_json(): """Test writing empty JSON""" data = {} output_file = WriteTester() script_output.write_json(output_file, data) lines = output_file.lines() assert len(lines) == 2 assert lines[0] == '[\n' assert lines[1] == '\n]\n'
def test_output_json(random1, random2): """Test to make sure the JSON is written out correctly""" data = [{'link': 'courage', 'zelda': 'wisdom', 'ganon': 'power', 'num': 3}, {'random_one': random1, 'random_two': random2}] output_file = WriteTester() script_output.write_json(output_file, data) lines = output_file.lines() assert len(lines) == 5 assert json.loads(''.join(lines)) assert lines[0] == '[\n' assert lines[2] == ',\n' assert lines[4] == '\n]\n' dict1 = json.loads(lines[1]) dict2 = json.loads(lines[3]) assert dict1['link'] == 'courage' assert dict1['zelda'] == 'wisdom' assert dict1['ganon'] == 'power' assert dict1['num'] == 3 assert dict2['random_one'] == random1 assert dict2['random_two'] == random2