def test_custom_schema_roundtrip_on_base_class(self): class Dummy(ResultHandler): def read(self, *args, **kwargs): pass def write(self, *args, **kwargs): pass schema = CustomResultHandlerSchema() obj = schema.load(schema.dump(Dummy())) assert obj is None
def test_custom_schema_roundtrip_on_base_class(self): class Dummy(ResultHandler): def read(self, *args, **kwargs): pass def write(self, *args, **kwargs): pass schema = CustomResultHandlerSchema() obj = schema.load(schema.dump(Dummy())) assert isinstance(obj, ResultHandler) assert obj.read(42) is None # just the base class, not the Dummy class
def test_custom_schema_roundtrip_on_stateful_class(self): class Stateful(ResultHandler): def __init__(self, x): self.x = x def read(self, *args, **kwargs): pass def write(self, *args, **kwargs): pass schema = CustomResultHandlerSchema() obj = schema.load(schema.dump(Stateful(42))) assert obj is None
def test_custom_schema_roundtrip_on_stateful_class(self): class Stateful(ResultHandler): def __init__(self, x): self.x = x def read(self, *args, **kwargs): pass def write(self, *args, **kwargs): pass schema = CustomResultHandlerSchema() obj = schema.load(schema.dump(Stateful(42))) assert isinstance(obj, ResultHandler) assert obj.write("foo") is None # just the base class, not the Stateful class
def test_custom_schema_dump_on_dummy_class(self): class Dummy(ResultHandler): def read(self, *args, **kwargs): pass def write(self, *args, **kwargs): pass serialized = CustomResultHandlerSchema().dump(Dummy()) assert serialized["type"].endswith("Dummy") assert serialized["__version__"] == prefect.__version__
def test_custom_schema_dump_on_stateful_class(self): class Stateful(ResultHandler): def __init__(self, x): self.x = x def read(self, *args, **kwargs): pass def write(self, *args, **kwargs): pass serialized = CustomResultHandlerSchema().dump(Stateful(42)) assert serialized["type"].endswith("Stateful") assert serialized["__version__"] == prefect.__version__