Should be:

```python
from unittest import TestCase, main
```