def final_node_query(self, node_index):
col = db.get_column(node_index)
paths = []
for path_index, t in col.items():
if t[0] == (t[1] - 1):
paths.append(db.get_path(path_index))
return paths
def test_read_sparse_matrix_column_returns_dict_of_size_three_lists():
db.clear_sparse_matrix()
db.store_tuple(0, 0, [1, 2, 3])
db.store_tuple(2, 0, [2, 3, 4])
col = db.get_column(0)
assert isinstance(col, dict)
assert isinstance(col[0], list)
with pytest.raises(KeyError):
col[1]
assert isinstance(col[2], list)
assert 3 == len(col[0])
assert 3 == len(col[2])
def node_query(self, node_index):
col = db.get_column(node_index)
paths = []
for path_index in col.keys():
paths.append(db.get_path(path_index))
return paths
def test_read_sparse_matrix_column_out_of_range_returns_None():
db.clear_sparse_matrix()
assert None == db.get_column(3)
