def test_dict_children_container(): # test to_data cc = core.DictChildrenContainer({}) nt.assert_equal({}, cc.to_data()) # test children_container_to_data as_data = core.children_container_to_data(cc) nt.assert_equal(dict(children_container_key="dict", children_container_data={}), as_data) # test back and forth cc2 = core.children_container_from_data(as_data) nt.assert_is_instance(cc2, core.DictChildrenContainer) nt.assert_equal(as_data, core.children_container_to_data(cc2))
def test_list_children_container(): # test to_data cc = core.ListChildrenContainer([]) nt.assert_equal([], cc.to_data()) # test children_container_to_data as_data = core.children_container_to_data(cc) nt.assert_equal(dict(children_container_key="list", children_container_data=[]), as_data) # test back and forth cc2 = core.children_container_from_data(as_data) nt.assert_is_instance(cc2, core.ListChildrenContainer) nt.assert_equal(as_data, core.children_container_to_data(cc2))
def test_nodes_and_edges_children_container(): # test to_data cc = core.NodesAndEdgesContainer([[], []]) nt.assert_equal({"nodes": [], "edges": []}, cc.to_data()) # test children_container_to_data as_data = core.children_container_to_data(cc) nt.assert_equal( dict(children_container_key="nodes_and_edges", children_container_data={"nodes": [], "edges": []}), as_data ) # test back and forth cc2 = core.children_container_from_data(as_data) nt.assert_is_instance(cc2, core.NodesAndEdgesContainer) nt.assert_equal(as_data, core.children_container_to_data(cc2))
def test_list_children_container(): # test to_data cc = core.ListChildrenContainer([]) nt.assert_equal([], cc.to_data()) # test children_container_to_data as_data = core.children_container_to_data(cc) nt.assert_equal( dict( children_container_key="list", children_container_data=[], ), as_data) # test back and forth cc2 = core.children_container_from_data(as_data) nt.assert_is_instance(cc2, core.ListChildrenContainer) nt.assert_equal(as_data, core.children_container_to_data(cc2))
def test_dict_children_container(): # test to_data cc = core.DictChildrenContainer({}) nt.assert_equal({}, cc.to_data()) # test children_container_to_data as_data = core.children_container_to_data(cc) nt.assert_equal( dict( children_container_key="dict", children_container_data={}, ), as_data) # test back and forth cc2 = core.children_container_from_data(as_data) nt.assert_is_instance(cc2, core.DictChildrenContainer) nt.assert_equal(as_data, core.children_container_to_data(cc2))
def test_dict_children_container_schema_serialization(): dccs = core.DictChildrenContainerSchema(foo=core.ListChildrenContainer, bar=core.ChildContainer) node = tn.AddConstantNode("hello") in_map = {"foo": [node, node], "bar": node} cc1 = dccs(in_map) cc2 = core.children_container_from_data(core.children_container_to_data(cc1)) nt.assert_equal(cc1.__class__, cc2.__class__) nt.assert_equal(cc1.__dict__, cc2.__dict__)
def test_nodes_and_edges_children_container(): # test to_data cc = core.NodesAndEdgesContainer([[], []]) nt.assert_equal({"nodes": [], "edges": []}, cc.to_data()) # test children_container_to_data as_data = core.children_container_to_data(cc) nt.assert_equal( dict(children_container_key="nodes_and_edges", children_container_data={ "nodes": [], "edges": [] }), as_data) # test back and forth cc2 = core.children_container_from_data(as_data) nt.assert_is_instance(cc2, core.NodesAndEdgesContainer) nt.assert_equal(as_data, core.children_container_to_data(cc2))
def test_dict_children_container_schema_serialization(): dccs = core.DictChildrenContainerSchema( foo=core.ListChildrenContainer, bar=core.ChildContainer, ) node = tn.AddConstantNode("hello") in_map = {"foo": [node, node], "bar": node} cc1 = dccs(in_map) cc2 = core.children_container_from_data( core.children_container_to_data(cc1)) nt.assert_equal(cc1.__class__, cc2.__class__) nt.assert_equal(cc1.__dict__, cc2.__dict__)