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)