Merge pull request python-quantities#186 from scidash/master

apdavison · web-flow · commit 269d81e8f1fa · 2021-10-18T10:26:05.000+02:00
Fixed non-list tolist() bug
diff --git a/quantities/quantity.py b/quantities/quantity.py
@@ -446,7 +446,10 @@ def tolist(self):
         #first get a dummy array from the ndarray method
         work_list = self.magnitude.tolist()
         #now go through and replace all numbers with the appropriate Quantity
-        self._tolist(work_list)
+        if isinstance(work_list, list):
+            self._tolist(work_list)
+        else:
+            work_list = Quantity(work_list, self.dimensionality)
         return work_list
 
     def _tolist(self, work_list):
diff --git a/quantities/tests/test_methods.py b/quantities/tests/test_methods.py
@@ -9,6 +9,8 @@ def setUp(self):
 
     def test_tolist(self):
         self.assertEqual(self.q.tolist(), [[1*pq.m, 2*pq.m], [3*pq.m, 4*pq.m]])
+        q_singleton = 1 * pq.m
+        self.assertEqual(q_singleton.tolist(), q_singleton)
 
     def test_sum(self):
         self.assertQuantityEqual(self.q.sum(), 10*pq.m)
diff --git a/quantities/tests/test_umath.py b/quantities/tests/test_umath.py
@@ -152,15 +152,12 @@ def test_ceil(self):
             )
 
     def test_fix(self):
-        try:
-            self.assertQuantityEqual(np.fix(3.14 * pq.degF), 3.0 * pq.degF)
-            self.assertQuantityEqual(np.fix(3.0 * pq.degF), 3.0 * pq.degF)
-            self.assertQuantityEqual(
-                np.fix([2.1, 2.9, -2.1, -2.9] * pq.degF),
-                [2., 2., -2., -2.] * pq.degF
-                )
-        except ValueError as e:
-            raise self.failureException(e)
+        self.assertQuantityEqual(np.fix(3.14 * pq.degF), 3.0 * pq.degF)
+        self.assertQuantityEqual(np.fix(3.0 * pq.degF), 3.0 * pq.degF)
+        self.assertQuantityEqual(
+            np.fix([2.1, 2.9, -2.1, -2.9] * pq.degF),
+            [2., 2., -2., -2.] * pq.degF
+            )
 
     def test_exp(self):
         self.assertQuantityEqual(np.exp(1*pq.dimensionless), np.e)
