def test_write_to_disk(self): tl = self.test_template_labels_assimilate_and_get_map() tl.write_to_disk(BASE_TEST_DATA_DIR) labels_filename = os.path.join(BASE_TEST_DATA_DIR, 'labels.dat') labels_dict = md.load_labels_file(labels_filename) os.remove(labels_filename) correct_dict = {1: 'aggregate', 2: 'boiler', 3: 'solar', 4: 'laptop', 5: 'washing_machine', 6: 'dishwasher', 7: 'tv', 8: 'kitchen_lights', 9: 'htpc', 10: 'kettle', 11: 'toaster', 12: 'fridge', 13: 'microwave', 14: 'lcd_office', 15: 'breadmaker', 16: 'hifi_office', 17: 'coffee', 18: 'amp_livingroom', 19: 'adsl_router'} self.assertEqual(labels_dict, correct_dict)
def test_load_labels_file(self): labels_filename = os.path.join(BASE_TEST_DATA_DIR, '001', 'labels.dat') l = md.load_labels_file(labels_filename) self.assertEqual(l[1], 'aggregate') self.assertEqual(l[17], 'amp_livingroom')