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))
