def test_invalid_weightings(self): weightings = _get_weightings('010003') self.assertEqual(len(weightings), 0) weightings = _get_weightings('049999') self.assertEqual(len(weightings), 0) weightings = _get_weightings('999999') self.assertEqual(len(weightings), 0)
def test_valid_weightings(self): weightings = _get_weightings('010005') self.assertEqual(len(weightings), 1) self.assertEqual(weightings[0].algorithm, 'MOD11') self.assertEqual(weightings[0].exception_code, None) self.assertEqual(weightings[0].weightings, [0, 0, 0, 0, 0, 0, 8, 7, 6, 5, 4, 3, 2, 1]) weightings = _get_weightings('070116') self.assertEqual(len(weightings), 2) self.assertEqual(weightings[0].algorithm, 'MOD11') self.assertEqual(weightings[0].exception_code, '12') self.assertEqual(weightings[1].algorithm, 'MOD10') weightings = _get_weightings('989999') self.assertEqual(len(weightings), 2) self.assertEqual(weightings[0].algorithm, 'MOD11') self.assertEqual(weightings[1].algorithm, 'DBLAL')