def test_network_from_pandas_hdf5( tmpfile, network, nodes, edges_df, impedance_names, two_way): ph5.network_to_pandas_hdf5(network, tmpfile) new_net = ph5.network_from_pandas_hdf5(Network, tmpfile) pdt.assert_frame_equal(new_net.nodes_df, nodes) pdt.assert_frame_equal(new_net.edges_df, edges_df) assert new_net._twoway == two_way assert new_net.impedance_names == impedance_names
def test_network_from_pandas_hdf5(tmpfile, network, nodes, edges_df, impedance_names, two_way): ph5.network_to_pandas_hdf5(network, tmpfile) new_net = ph5.network_from_pandas_hdf5(Network, tmpfile) pdt.assert_frame_equal(new_net.nodes_df, nodes) pdt.assert_frame_equal(new_net.edges_df, edges_df) assert new_net._twoway == two_way assert new_net.impedance_names == impedance_names
def test_network_to_pandas_hdf5( tmpfile, network, nodes, edges_df, impedance_names, two_way): ph5.network_to_pandas_hdf5(network, tmpfile) store = pd.HDFStore(tmpfile) pdt.assert_frame_equal(store['nodes'], nodes) pdt.assert_frame_equal(store['edges'], edges_df) 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(tmpfile, network, nodes, edges_df, impedance_names, two_way): ph5.network_to_pandas_hdf5(network, tmpfile) store = pd.HDFStore(tmpfile) pdt.assert_frame_equal(store['nodes'], nodes) pdt.assert_frame_equal(store['edges'], edges_df) 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_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))