fix SE2 constructor logic

petercorke · petercorke · commit 29dcacfab6d8 · 2020-09-07T18:34:30.000+10:00
diff --git a/spatialmath/pose2d.py b/spatialmath/pose2d.py
@@ -294,7 +294,8 @@ def __init__(self, x=None, y=None, theta=None, *, unit='rad', check=True):
             elif y is not None and theta is not None:
                 # SE2(x, y, theta)
                 self.data = [tr.trot2(theta, t=[x, y], unit=unit)]
-            elif y is None and theta is None:
+            elif y is None and theta is None and argcheck.isnumberlist(x):
+                # SE2(arg)
                 arg = argcheck.getvector(x)
                 if len(arg) == 1:
                     self.data = [tr.trot2(arg[0], unit=unit)]
@@ -306,6 +307,8 @@ def __init__(self, x=None, y=None, theta=None, *, unit='rad', check=True):
                     self.data = [tr.trot2(arg[2], t=arg[:2], unit=unit)]
                 else:
                     super().arghandler(x, check=check)
+            else:
+                self.arghandler(x, check=check)
         else:
             raise ValueError('bad arguments to constructor')
 
