Fix bug when doing get_data from an Array with one element. (shapelets#50)

Antonio Vilches · web-flow · commit 1c719c6cf2c9 · 2019-05-06T20:16:48.000+02:00
* Fix bug when doing get_data from an Array with one element.
diff --git a/khiva/array.py b/khiva/array.py
@@ -213,7 +213,11 @@ def _get_data(self):
         KhivaLibrary().c_khiva_library.get_data(ctypes.pointer(self.arr_reference), ctypes.pointer(c_result_array))
 
         dims = self.get_dims()
-        dims = dims[dims > 1]
+        if dims[dims > 1].size > 0:
+            dims = dims[dims > 1]
+        else:
+            dims = np.array([1])
+
         a = np.array(c_result_array)
 
         if self._is_complex():
diff --git a/tests/unit_tests/array_unit_tests.py b/tests/unit_tests/array_unit_tests.py
@@ -394,6 +394,15 @@ def testNonZero(self):
         a = Array([1, 2, 3, 4])
         self.assertTrue(a)
 
+    def testGetData1(self):
+        a = Array([1])
+        np.testing.assert_array_equal(a._get_data(), np.array([1]))
+
+    def testToList1(self):
+        a = Array([1])
+        self.assertIs(type(a.to_list()), list)
+        self.assertEqual(a.to_list()[0], 1)
+
 
 if __name__ == '__main__':
     suite = unittest.TestLoader().loadTestsFromTestCase(ArrayTest)
