class TestDK(unittest.TestCase): def setUp(self): self.validator = NinValidator(country='dk') def test_sanitize(self): self.validator.nin = '123456-1234' self.assertEqual(self.validator.sanitize, '1234561234') self.validator.nin = '123456 1234' self.assertEqual(self.validator.sanitize, '1234561234') self.validator.nin = '123456--1234' self.assertEqual(self.validator.sanitize, '1234561234') self.validator.nin = 1234561234 self.assertEqual(self.validator.sanitize, '1234561234') with self.assertRaises(TypeError): self.validator.nin = 'abc' self.validator.sanitize() with self.assertRaises(TypeError): self.validator.nin = '123456-123A' self.validator.sanitize() with self.assertRaises(TypeError): self.validator.nin = '123456+1234' self.validator.sanitize() def test_getage(self): diff = datetime.date.today().year - 1900 self.validator.nin = '100600-1234' self.assertEqual(self.validator.age, diff) diff = datetime.date.today().year - 1981 self.validator.nin = '100681-1234' self.assertEqual(self.validator.age, diff) diff = datetime.date.today().year - 1881 self.validator.nin = '100681-5234' self.assertEqual(self.validator.age, diff) diff = datetime.date.today().year - 2000 self.validator.nin = '100600-9234' self.assertEqual(self.validator.age, diff) diff = datetime.date.today().year - 2000 self.validator.nin = '100600-5234' self.assertEqual(self.validator.age, diff)
def setUp(self): self.validator = NinValidator(country='dk')