def assert_state(self, state_class: Type[BaseState]):
assert self._register<EOL>assert Register.STATE in self._register<EOL>assert self._register[Register.STATE] == state_class.name()<EOL>
Assert that the state returned in the register is the one passed as
argument.