def test_pickle_json_includes_class_attributes(self): @dataclass class TestEventClassAttributes(BaseEvent): type = 'ai.anagon.base' sub_type = 'ai.anagon.child' content: str self.assertEqual( { 'type': 'ai.anagon.base', 'sub_type': 'ai.anagon.child', 'content': 'test', }, Pickler(unpicklable=False).flatten( TestEventClassAttributes("test")))
def _get_serialized_context(state: BotState, context: TurnContext): ctx = state.get(context) return Pickler(unpicklable=False).flatten(ctx)
def print_json(self, event: MathParsed): print(Pickler().flatten(event))