def test_remove_nodes(network, rm_nodes):
# node 0 is connected to node 4, which is in turn connected to node 5
# node 7 is connected to node 11, which has no other connections
# node 6 is connected to nodes 2, 5, and 10,
# which all have other connections
nodes, edges = ph5.remove_nodes(network, rm_nodes)
exp_nodes = pd.DataFrame(
{
'x': [2, 3, 4, 1, 2, 1, 2, 3, 4],
'y': [1, 1, 1, 2, 2, 3, 3, 3, 3]
},
index=[1, 2, 3, 4, 5, 8, 9, 10, 11])
exp_edges = pd.DataFrame(
{
'from': [4, 2, 2, 10, 9],
'to': [5, 1, 3, 9, 8],
'distance': [1, 1, 1, 1, 1],
'time': [2, 5, 6, 8, 9]
},
index=[1, 4, 5, 7, 8])
exp_edges = exp_edges[['from', 'to', 'distance', 'time']] # order columns
pdt.assert_frame_equal(nodes, exp_nodes)
pdt.assert_frame_equal(edges, exp_edges)
def test_network_save_load_hdf5(
tmpfile, network, impedance_names, two_way, rm_nodes):
network.save_hdf5(tmpfile, rm_nodes)
new_net = Network.from_hdf5(tmpfile)
nodes, edges = ph5.remove_nodes(network, rm_nodes)
pdt.assert_frame_equal(new_net.nodes_df, nodes)
pdt.assert_frame_equal(new_net.edges_df, edges)
assert new_net._twoway == two_way
assert new_net.impedance_names == impedance_names
def test_network_save_load_hdf5(tmpfile, network, impedance_names, two_way,
rm_nodes):
network.save_hdf5(tmpfile, rm_nodes)
new_net = Network.from_hdf5(tmpfile)
nodes, edges = ph5.remove_nodes(network, rm_nodes)
pdt.assert_frame_equal(new_net.nodes_df, nodes)
pdt.assert_frame_equal(new_net.edges_df, edges)
assert new_net._twoway == two_way
assert new_net.impedance_names == impedance_names
def test_network_to_pandas_hdf5_removal(
tmpfile, network, impedance_names, two_way, rm_nodes):
nodes, edges = ph5.remove_nodes(network, rm_nodes)
ph5.network_to_pandas_hdf5(network, tmpfile, rm_nodes)
store = pd.HDFStore(tmpfile)
pdt.assert_frame_equal(store['nodes'], nodes)
pdt.assert_frame_equal(store['edges'], edges)
pdt.assert_series_equal(store['two_way'], pd.Series([two_way]))
pdt.assert_series_equal(
store['impedance_names'], pd.Series(impedance_names))
def test_network_to_pandas_hdf5_removal(tmpfile, network, impedance_names,
two_way, rm_nodes):
nodes, edges = ph5.remove_nodes(network, rm_nodes)
ph5.network_to_pandas_hdf5(network, tmpfile, rm_nodes)
store = pd.HDFStore(tmpfile)
pdt.assert_frame_equal(store['nodes'], nodes)
pdt.assert_frame_equal(store['edges'], edges)
pdt.assert_series_equal(store['two_way'], pd.Series([two_way]))
pdt.assert_series_equal(store['impedance_names'],
pd.Series(impedance_names))
def test_remove_nodes(network, rm_nodes):
# node 0 is connected to node 4, which is in turn connected to node 5
# node 7 is connected to node 11, which has no other connections
# node 6 is connected to nodes 2, 5, and 10,
# which all have other connections
nodes, edges = ph5.remove_nodes(network, rm_nodes)
exp_nodes = pd.DataFrame(
{'x': [2, 3, 4, 1, 2, 1, 2, 3, 4],
'y': [1, 1, 1, 2, 2, 3, 3, 3, 3]},
index=[1, 2, 3, 4, 5, 8, 9, 10, 11])
exp_edges = pd.DataFrame(
{'from': [4, 2, 2, 10, 9],
'to': [5, 1, 3, 9, 8],
'distance': [1, 1, 1, 1, 1],
'time': [2, 5, 6, 8, 9]},
index=[1, 4, 5, 7, 8])
exp_edges = exp_edges[['from', 'to', 'distance', 'time']] # order columns
pdt.assert_frame_equal(nodes, exp_nodes)
pdt.assert_frame_equal(edges, exp_edges)
