def test_write(self): document = {'Disco': {1: 'Beatdown', 2: 'Elysium'}} yaml_file_interface.write('./tmp/test_write.yaml', document) self.assertTrue( filecmp.cmp('./tmp/test_write.yaml', self.get_static_file('test_write.yaml')), 'Files do not match')
def test_write_empty_file(self): document = {} yaml_path = './tmp/test_write_empty_file.yaml' yaml_file_interface.write(yaml_path, document) self.assertTrue( filecmp.cmp(yaml_path, self.get_static_file('test_write_empty_file.yaml')), 'Files do not match')
def test_write_list_contents(self): document = ['Disco', 'Fever'] yaml_path = './tmp/test_write_list_contents.yaml' yaml_file_interface.write(yaml_path, document) # A list should just translate to a linear series of YAML keys self.assertTrue( filecmp.cmp(yaml_path, self.get_static_file('test_write_list_contents.yaml')), 'Files do not match')
def test_write_overwrite(self): document = {'Disco': {1: 'Beatdown', 2: 'Elysium'}} yaml_file_interface.write('./tmp/test_write_overwrite.yaml', document) self.assertEqual(len(document['Disco']), 2, 'Number of entries is incorrect') document['Disco'][3] = 'Fever' yaml_file_interface.write('./tmp/test_write_overwrite.yaml', document) self.assertEqual(document['Disco'][3], 'Fever', 'Third entry is incorrect')
def test_write_string_contents(self): document = 'Ugh' yaml_path = './tmp/test_write_string_contents.yaml' yaml_file_interface.write(yaml_path, document) # Despite not being a dictionary, this writes valid YAML to the file - the result being a single YAML key self.assertTrue( filecmp.cmp( yaml_path, self.get_static_file('test_write_string_contents.yaml')), 'Files do not match')