Merge pull request #333 from jcrist/pluck-default-test

Add tests for pluck with default, non-None
pytoolz · Oct 3, 2016 · 068d1d9 · 068d1d9
2 parents 68ede01 + 9ff94a3
commit 068d1d9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/toolz/tests/test_itertoolz.py b/toolz/tests/test_itertoolz.py
@@ -335,11 +335,10 @@ def test_pluck():
 
     data = [{'id': 1, 'name': 'cheese'}, {'id': 2, 'name': 'pies', 'price': 1}]
     assert list(pluck('id', data)) == [1, 2]
-    assert list(pluck('price', data, None)) == [None, 1]
+    assert list(pluck('price', data, 0)) == [0, 1]
     assert list(pluck(['id', 'name'], data)) == [(1, 'cheese'), (2, 'pies')]
     assert list(pluck(['name'], data)) == [('cheese',), ('pies',)]
-    assert list(pluck(['price', 'other'], data, None)) == [(None, None),
-                                                           (1, None)]
+    assert list(pluck(['price', 'other'], data, 0)) == [(0, 0), (1, 0)]
 
     assert raises(IndexError, lambda: list(pluck(1, [[0]])))
     assert raises(KeyError, lambda: list(pluck('name', [{'id': 1}])))