Fix tests
Auto-import the relevant classes
Fix inheritance and remove local __repr__ implementations where they are already inherited from the parent