Merge pull request #165 from InsightLab/developer

Bump version: 2.6.0 → 2.6.1

Author: mxaviersmp

.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 2.6.0
+current_version = 2.6.1
 allow_dirty = True
 tag_name = version-{new_version}
 tag = True

pymove/_version.py

@@ -1 +1 @@
-__version__ = '2.6.0'
+__version__ = '2.6.1'

pymove/tests/test_utils_trajectories.py

@@ -96,7 +96,17 @@ def test_shift():
 
     expected = [4, 5, 0, 0, 0]
     array_ = [1, 2, 3, 4, 5]
-    shifted_array = trajectories.shift(arr=array_, num=-3, fill_value=0)
+    shifted_array = trajectories.shift(arr=array_, num=-3)
+    assert_array_equal(shifted_array, expected)
+
+    expected = [False, False, False, True, True]
+    array_ = [True, True, True, True, True]
+    shifted_array = trajectories.shift(arr=array_, num=3)
+    assert_array_equal(shifted_array, expected)
+
+    expected = ['dewberry', 'eggplant', 'nan', 'nan', 'nan']
+    array_ = ['apple', 'banana', 'coconut', 'dewberry', 'eggplant']
+    shifted_array = trajectories.shift(arr=array_, num=-3, fill_value=np.nan)
     assert_array_equal(shifted_array, expected)
 
 

pymove/utils/trajectories.py

@@ -210,7 +210,10 @@ def shift(
 
     result = np.empty_like(arr)
     if fill_value is None:
-        if isinstance(result.dtype, int):
+        dtype = result.dtype
+        if np.issubdtype(dtype, np.bool_):
+            fill_value = False
+        elif np.issubdtype(dtype, np.integer):
             fill_value = 0
         else:
             fill_value = np.nan

setup.py

@@ -33,7 +33,7 @@
 ]
 setup(
     name='pymove',
-    version='2.6.0',
+    version='2.6.1',
     author='Insight Data Science Lab',
     author_email='[email protected]',
     license='MIT',