def test_read_sparse_matrix_row_returns_dict_of_size_three_lists():
db.clear_sparse_matrix()
db.store_tuple(0, 0, [1, 2, 3])
db.store_tuple(0, 2, [2, 3, 4])
row = db.get_row(0)
assert isinstance(row, dict)
assert isinstance(row[0], list)
with pytest.raises(KeyError):
row[1]
assert isinstance(row[2], list)
assert 3 == len(row[0])
assert 3 == len(row[2])
def path_query(self, path_index):
row = db.get_row(path_index)
nodes = []
for node_index in row.keys():
nodes.append(db.get_node(node_index))
return nodes
def test_read_sparse_matrix_row_out_of_range_returns_None():
db.clear_sparse_matrix()
assert None == db.get_row(3)
