예제 #1
0
    def test_read_team_from_file(self):

        doc = team._read_team_from_file(self.testfile)

        self.assertIsNotNone(doc['team'])
        self.assertIsNotNone(doc['team']['alice'])
        self.assertIsNotNone(doc['team']['bob'])
        self.assertIsNotNone(doc['team']['chris'])
        self.assertIsNotNone(doc['team']['debby'])
        self.assertIsNotNone(doc['holidays'])
        self.assertIsNotNone(doc['holidays'])
예제 #2
0
    def test_init_team(self):

        doc = team._read_team_from_file(self.testfile)
        result = team._init_team(doc)

        self.assertIsNotNone(result)
        self.assertEqual(len(result), 5)

        for member in result:
            if member.name == 'alice':
                self.assertEqual(len(member.unavail['default']), 2)
                self.assertIsNotNone(member.holidays)
            if member.name == 'bob':
                self.assertEqual(len(member.unavail['odd_week']), 2)
                self.assertIsNotNone(member.holidays)