def assert_infer_state(self, expected): state = pinfer.format_state() self.assertEqual(state, expected) pinfer.reset()