def test_set_line_graph_data(self):
branch = self.make_test_branch(
'A', message='badger', committer='fnord')
rev_id = branch.repository.get_revision('A').revision_id
model = BranchTreeModel(branch, [])
data = [(rev_id, (2, 0), (3, 4, 0.5), None, None, [1, 2])]
model.set_line_graph_data(data)
self.assertEqual(data, model.line_graph_data)
tree_iter = model.get_iter_first()
self.assertEqual(rev_id, model.get_value(tree_iter, 0))
