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))