While we don't do it yet (we probably should), `pytest` also supports tests organized into classes. These can be `unittest.TestCase` subclasses, or just bare classes that contain test functions, so we should make sure to import those as well. I guess that would involve another function that inspects a class and strips out methods that don't contain the `Simulator` arg?