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__
