Simplify test_comment.

A. Jesse Jiryu Davis · A. Jesse Jiryu Davis · commit 85ad5bd8dd80 · 2013-12-14T13:01:45.000-05:00
diff --git a/test/test_cursor.py b/test/test_cursor.py
@@ -953,7 +953,7 @@ def run_with_profiling(func):
             self.db.set_profiling_level(OFF)
 
         def find():
-            list(self.db.test.find({'type': 'string'}).comment('foo'))
+            list(self.db.test.find().comment('foo'))
             op = self.db.system.profile.find({'ns': 'pymongo_test.test',
                                               'op': 'query',
                                               'query.$comment': 'foo'})
@@ -962,7 +962,7 @@ def find():
         run_with_profiling(find)
 
         def count():
-            self.db.test.find({'type': 'string'}).comment('foo').count()
+            self.db.test.find().comment('foo').count()
             op = self.db.system.profile.find({'ns': 'pymongo_test.$cmd',
                                               'op': 'command',
                                               'command.count': 'test',
@@ -972,7 +972,7 @@ def count():
         run_with_profiling(count)
 
         def distinct():
-            self.db.test.find({'type': 'string'}).comment('foo').distinct('type')
+            self.db.test.find().comment('foo').distinct('type')
             op = self.db.system.profile.find({'ns': 'pymongo_test.$cmd',
                                               'op': 'command',
                                               'command.distinct': 'test',
