[WIP] Improve python test coverage

Is there a reason to `setUpModule` instead of `setUpClass` inside the `TestCase`? Do you intend to add more `TestCase`s?