wip

layercake/tests/test_funcs.py

@@ -0,0 +1,21 @@
+from layercake.funcs import omit, pick
+
+
+def test_omit():
+    values = {'indigo': '#4b0082', 'navy': '#000080'}
+    assert omit(['indigo'], values) == {'navy': '#000080'}
+    assert omit(['test'], values) == values
+
+
+def test_pick():
+    values = {'indigo': '#4b0082', 'navy': '#000080'}
+    assert pick(['navy'], values) == {'navy': '#000080'}
+    assert pick(['test'], values) == {}
+
+
+def test_pick_default_val():
+    values = {'name': 'test'}
+    assert pick(['name', 'surname'], values, exclude_none=False, default=False) == {
+        'name': 'test',
+        'surname': False,
+    }