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')
