def export_excell(self):
filename = str(QtGui.QFileDialog.getSaveFileName(self, 'Export population Excell File', '.xls'))
if filename:
saver = PopulationSavingObject(self._institution.people, filename)
saver.create()
saver.populate()
filename = str(QtGui.QFileDialog.getSaveFileName(self, 'Export duration Excell File', '.xls'))
if filename:
saver = DurationSavingObject(self.durations_widget.singleDurationWidget.duration, filename)
saver.create()
saver.populate()
def testSaveLoadPopulation(self):
self.institution.people.append(new_worker())
saver = PopulationSavingObject(self.institution.people, 'temp_sheet.xls')
saver.create()
saver.populate()
loader = PopulationSavingObject([], 'temp_sheet.xls')
loaded_population = loader.load()
self.assertEqual(len(loaded_population), 1)
loaded_person = loaded_population[0]
self.assertEqual(loaded_person.name, 'Bob')
self.assertEqual(loaded_person.address, 'a')
self.assertEqual(loaded_person.email, 'b')
self.assertEqual(loaded_person.phone_number, '0115')
self.assertEqual(len(loaded_person.roles()), 2)
self.assertTrue(loaded_person.suitable_for_role('B'))
self.assertTrue(loaded_person.suitable_for_role('S'))
self.assertFalse(loaded_person.suitable_for_role('F'))
def testSavePopulation(self):
self.institution.people.append(new_worker())
saver = PopulationSavingObject(self.institution.people, 'temp_sheet.xls')
saver.create()
saver.populate()
