tests with start values

Author: papajohn

datascience/util.py

@@ -130,6 +130,8 @@ def minimize(f, start=None, **vargs):
         arg_count = f.__code__.co_argcount
         assert arg_count > 0, "Please pass starting values explicitly"
         start = [0] * arg_count
+    if not hasattr(start, '__len__'):
+        start = [start]
 
     @functools.wraps(f)
     def wrapper(args):

tests/test_util.py

@@ -44,3 +44,5 @@ def test_table_apply():
 def test_minimize():
     assert (2 == ds.minimize(lambda x: (x-2)**2)) == True
     assert [2, 1] == list(ds.minimize(lambda x, y: (x-2)**2 + (y-1)**2))
+    assert (2 == ds.minimize(lambda x: (x-2)**2, 1)) == True
+    assert [2, 1] == list(ds.minimize(lambda x, y: (x-2)**2 + (y-1)**2, [1, 1]))