Allow empty y in new vector

2020-02-24 17:46:30 -08:00 · 2020-02-24 17:46:30 -08:00 · e2e582ab03
parent 6a4cdaab90
commit e2e582ab03
1 changed files with 2 additions and 2 deletions
--- a/vector.py
+++ b/vector.py
@ -5,9 +5,9 @@ def vec_or_num(x, y = None):
    return x

 class Vector:
-    def __init__(self, x = 0, y = 0):
+    def __init__(self, x = 0, y = None):
        self.x = float(x)
-        self.y = float(y)
+        self.y = float(y or x)

    def __repr__(self):
        return "Vector {{x: {0}, y: {1}}}".format(self.x, self.y)