class TimesheetTestCase(unittest.TestCase): def setUp(self): self.timesheet = TimesheetManager(os.path.join(root, "tests/example.yaml")) def test_meta(self): self.assertEqual(self.timesheet.get_project_name(), "Example Project") def test_entries(self): self.assertEqual(self.timesheet.get_total_entries(), 3) def test_total_hours(self): self.assertEqual(self.timesheet.get_total_hours(), 6.5) def test_hours_within(self): f = datetime.datetime(2012, 7, 1) t = datetime.datetime(2012, 8, 1) self.assertEqual(self.timesheet.get_hours_within(f, t), 6.5) f = datetime.datetime(2012, 7, 1) t = datetime.datetime(2012, 7, 31) self.assertEqual(self.timesheet.get_hours_within(f, t), 3.0) def test_add_hours(self): f = datetime.datetime(2012, 8, 1, 10, 0) t = datetime.datetime(2012, 8, 1, 12, 0) self.timesheet.add_hours(f, t, desc="Testing") self.assertEqual(self.timesheet.get_total_entries(), 4) f = datetime.datetime(2012, 8, 1) t = datetime.datetime(2012, 8, 31) self.assertEqual(self.timesheet.get_hours_within(f, t), 2.0)
def setUp(self): self.timesheet = TimesheetManager(os.path.join(root, "tests/example.yaml"))