Skip to content

Commit 1bfe5b8

Browse files
committed
New M/R tests PYTHON-239
1 parent 50e0cbd commit 1bfe5b8

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

test/test_collection.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1117,6 +1117,19 @@ def test_map_reduce(self):
11171117
merge_output=True,
11181118
reduce_output=True)
11191119

1120+
result = db.test.map_reduce(map, reduce, out={'replace': 'mrunittests'})
1121+
self.assertEqual(3, result.find_one({"_id": "cat"})["value"])
1122+
self.assertEqual(2, result.find_one({"_id": "dog"})["value"])
1123+
self.assertEqual(1, result.find_one({"_id": "mouse"})["value"])
1124+
1125+
result = db.test.map_reduce(map, reduce,
1126+
out=SON([('replace', 'mrunittests'),
1127+
('db', 'mrtestdb')
1128+
]))
1129+
self.assertEqual(3, result.find_one({"_id": "cat"})["value"])
1130+
self.assertEqual(2, result.find_one({"_id": "dog"})["value"])
1131+
self.assertEqual(1, result.find_one({"_id": "mouse"})["value"])
1132+
11201133
full_result = db.test.map_reduce(map, reduce,
11211134
out='mrunittests', full_response=True)
11221135
self.assertEqual(6, full_result["counts"]["emit"])
@@ -1127,6 +1140,13 @@ def test_map_reduce(self):
11271140
self.assertEqual(None, result.find_one({"_id": "mouse"}))
11281141

11291142
if version.at_least(self.db.connection, (1, 7, 4)):
1143+
result = db.test.map_reduce(map, reduce, out={'inline': 1})
1144+
self.assertTrue(isinstance(result, dict))
1145+
self.assertTrue('results' in result)
1146+
self.assertTrue(result['results'][1]["_id"] in ("cat",
1147+
"dog",
1148+
"mouse"))
1149+
11301150
result = db.test.inline_map_reduce(map, reduce)
11311151
self.assertTrue(isinstance(result, list))
11321152
self.assertEqual(3, len(result))

0 commit comments

Comments
 (0)