def test_2(self):
"""
Tests asymmetric string x string matrix
writing to and reading from .tmx.
"""
interface = MatrixInterface()
interface.primary_ids_are_string = True
interface.secondary_ids_are_string = True
interface.prepare_matrix(is_symmetric=False,
is_compressible=False,
rows=3,
columns=2,
network_vertices=4)
from_column = [0, 1, 0, 3, 0]
to_column = [1, 0, 3, 2, 2]
weight_column = [3, 4, 5, 7, 2]
is_bidirectional_column = [False, False, False, False, True]
interface.add_edges_to_graph(
from_column=from_column,
to_column=to_column,
edge_weight_column=weight_column,
is_bidirectional_column=is_bidirectional_column)
interface.add_user_source_data(2, "a", 5, False)
interface.add_user_source_data(1, "b", 4, False)
interface.add_user_source_data(0, "c", 1, False)
interface.add_user_dest_data(0, "d", 4)
interface.add_user_dest_data(3, "e", 6)
interface.build_matrix()
interface.add_to_category_map("d", "cat_a")
interface.add_to_category_map("e", "cat_b")
assert interface.get_dests_in_range(100) == {
"a": ["d", "e"],
"b": ["d", "e"],
"c": ["d", "e"]
}
filename = self.datapath + "test_1.tmx"
interface.write_tmx(filename)
interface2 = MatrixInterface()
interface2.read_file(filename)
interface2.add_to_category_map("d", "cat_a")
interface2.add_to_category_map("e", "cat_b")
assert interface2.get_dests_in_range(100) == {
"a": ["d", "e"],
"b": ["d", "e"],
"c": ["d", "e"]
}
interface2.write_csv(self.datapath + "test_2.csv")
def test_01(self):
"""
Tests asymmetric int x int matrix
writing to and reading from tmx.
"""
interface = MatrixInterface()
interface.prepare_matrix(is_symmetric=False,
is_compressible=False,
rows=3,
columns=2,
network_vertices=4)
from_column = [0, 1, 0, 3, 0]
to_column = [1, 0, 3, 2, 2]
weight_column = [3, 4, 5, 7, 2]
is_bidirectional_column = [False, False, False, False, True]
interface.add_edges_to_graph(
from_column=from_column,
to_column=to_column,
edge_weight_column=weight_column,
is_bidirectional_column=is_bidirectional_column)
interface.add_user_source_data(2, 10, 5, False)
interface.add_user_source_data(1, 11, 4, False)
interface.add_user_source_data(0, 12, 1, False)
interface.add_user_dest_data(0, 21, 4)
interface.add_user_dest_data(3, 20, 6)
interface.build_matrix()
interface.add_to_category_map(20, "a")
interface.add_to_category_map(21, "b")
assert interface.get_dests_in_range(100) == {
10: [21, 20],
11: [21, 20],
12: [21, 20]
}
filename = self.datapath + "test_1.tmx"
interface.write_tmx(filename)
interface2 = MatrixInterface()
interface2.read_file(filename)
interface2.add_to_category_map(20, "a")
interface2.add_to_category_map(21, "b")
assert interface.get_dests_in_range(100) == {
10: [21, 20],
11: [21, 20],
12: [21, 20]
}
interface2.write_csv(self.datapath + "test_1.csv")