You could remove the ```def main``` function and call ```unittest.main()``` directly here.