react-python-tdd-tutorial/exemples/4-bases-python-tdd/5-tdd.py

import unittest

def is_major(age):
    if type(age) is not int:
        age = int(age)
    return True if age >= 18 else False

class TestIsMajor(unittest.TestCase):

    def test_is_major(self):
        self.assertTrue(is_major(30))
        self.assertFalse(is_major(11))
        self.assertTrue(is_major("18"))
        self.assertFalse(is_major("17"))
        self.assertRaises(ValueError, is_major, 'bla')

if __name__ == '__main__':
    unittest.main()