Commit a7311523 authored by Bram Schoenmakers's avatar Bram Schoenmakers

Use the source() function.

parent e8e03f5f
...@@ -28,7 +28,7 @@ class TodoListTester(unittest.TestCase): ...@@ -28,7 +28,7 @@ class TodoListTester(unittest.TestCase):
count = self.todolist.count() count = self.todolist.count()
self.todolist.add(text) self.todolist.add(text)
self.assertEquals(self.todolist.todo(count+1).src, text) self.assertEquals(self.todolist.todo(count+1).source(), text)
self.assertEquals(set(['Project1', 'Project2', 'Project3']), \ self.assertEquals(set(['Project1', 'Project2', 'Project3']), \
self.todolist.projects()) self.todolist.projects())
self.assertEquals(set(['Context1', 'Context2', 'Context3']), \ self.assertEquals(set(['Context1', 'Context2', 'Context3']), \
...@@ -43,7 +43,7 @@ class TodoListTester(unittest.TestCase): ...@@ -43,7 +43,7 @@ class TodoListTester(unittest.TestCase):
count = self.todolist.count() count = self.todolist.count()
self.todolist.delete(2) self.todolist.delete(2)
self.assertEquals(self.todolist.todo(2).src, \ self.assertEquals(self.todolist.todo(2).source(), \
"(C) Baz @Context1 +Project1 key:value") "(C) Baz @Context1 +Project1 key:value")
self.assertEquals(self.todolist.count(), count - 1) self.assertEquals(self.todolist.count(), count - 1)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment