def test_can_save_and_load(tmpdir) -> None: PROTO_FILE_NAME = str(tmpdir.join("test.proto")) JSON_FILE_NAME = str(tmpdir.join("test.json")) DOT_FILE_NAME = str(tmpdir.join("test.dot")) gamma = Gamma(1.0, 1.0) gamma.set_value(2.5) # %%SNIPPET_START%% PythonSaveSnippet net = BayesNet(gamma.iter_connected_graph()) metadata = {"Author": "Documentation Team"} protobuf_saver = ProtobufSaver(net) protobuf_saver.save(PROTO_FILE_NAME, True, metadata) json_saver = JsonSaver(net) json_saver.save(JSON_FILE_NAME, True, metadata) dot_saver = DotSaver(net) dot_saver.save(DOT_FILE_NAME, True, metadata) # %%SNIPPET_END%% PythonSaveSnippet # %%SNIPPET_START%% PythonLoadSnippet protobuf_loader = ProtobufLoader() new_net_from_proto = protobuf_loader.load(PROTO_FILE_NAME) json_loader = JsonLoader() new_net_from_json = json_loader.load(JSON_FILE_NAME)
def test_can_save_and_load(tmpdir) -> None: PROTO_FILE = str(tmpdir.join("test.proto")) JSON_FILE = str(tmpdir.join("test.json")) DOT_FILE = str(tmpdir.join("test.dot")) gamma = Gamma(1.0, 1.0) gamma.set_value(2.5) net = BayesNet(gamma.get_connected_graph()) metadata = {"Team": "GraphOS"} protobuf_saver = ProtobufSaver(net) protobuf_saver.save(PROTO_FILE, True, metadata) json_saver = JsonSaver(net) json_saver.save(JSON_FILE, True, metadata) dot_saver = DotSaver(net) dot_saver.save(DOT_FILE, True, metadata) check_dot_file(DOT_FILE) protobuf_loader = ProtobufLoader() json_loader = JsonLoader() new_net_from_proto = protobuf_loader.load(PROTO_FILE) check_loaded_net(new_net_from_proto) new_net_from_json = json_loader.load(JSON_FILE) check_loaded_net(new_net_from_json)