def test_get_modal_network(request):
print("\n--Starting:", request.node.name)
mode = "transit"
print("Reading network. Mode: {} ...".format(mode))
net = RoadwayNetwork.read(
link_file=STPAUL_LINK_FILE,
node_file=STPAUL_NODE_FILE,
shape_file=STPAUL_SHAPE_FILE,
fast=True,
)
_links_df, _nodes_df = RoadwayNetwork.get_modal_links_nodes(
net.links_df, net.nodes_df, mode=mode,
)
mode_variable = RoadwayNetwork.MODES_TO_NETWORK_LINK_VARIABLES[mode]
non_transit_links =_links_df[_links_df[mode_variable] != 1]
assert non_transit_links.shape[0] == 0
def test_get_modal_network(request):
print("\n--Starting:", request.node.name)
mode = "transit"
print("Reading network. Mode: {} ...".format(mode))
net = RoadwayNetwork.read(
link_file=STPAUL_LINK_FILE,
node_file=STPAUL_NODE_FILE,
shape_file=STPAUL_SHAPE_FILE,
fast=True,
)
_links_df, _nodes_df = RoadwayNetwork.get_modal_links_nodes(
net.links_df,
net.nodes_df,
modes=[mode],
)
test_links_of_selection = _links_df["model_link_id"].tolist()
print("TEST - Number of selected links: {}".format(
len(test_links_of_selection)))
mode_variables = RoadwayNetwork.MODES_TO_NETWORK_LINK_VARIABLES[mode]
control_links_of_selection = []
for m in mode_variables:
control_links_of_selection.extend(net.links_df.loc[net.links_df[m],
"model_link_id"])
print("CONTROL - Number of selected links: {}".format(
len(control_links_of_selection)))
all_model_link_ids = _links_df["model_link_id"].tolist()
print("CONTROL - Number of total links: {}".format(
len(all_model_link_ids)))
assert set(test_links_of_selection) == set(control_links_of_selection)