diff mbox series

[kms-tests,2/6] kmstest: Add additional geometry classes

Message ID 20200807232119.28854-3-laurent.pinchart@ideasonboard.com (mailing list archive)
State New
Delegated to: Kieran Bingham
Headers show
Series Improve CRC (DISCOM) test | expand

Commit Message

Laurent Pinchart Aug. 7, 2020, 11:21 p.m. UTC 
Add Dist, Point and Size classes in addition to the Rect class.

Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
---
 tests/kmstest.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
diff mbox series

Patch

diff --git a/tests/kmstest.py b/tests/kmstest.py
index 0281c6727271..949bb20b8b1a 100755
--- a/tests/kmstest.py
+++ b/tests/kmstest.py
@@ -205,6 +205,37 @@  class CRCReader(object):
         return crcs
 
 
+class Dist(object):
+    def __init__(self, x, y):
+        self.x = x
+        self.y = y
+
+    def __repr__(self):
+        return "(%d,%d)" % (self.x, self.y)
+
+
+class Point(object):
+    def __init__(self, x, y):
+        self.x = x
+        self.y = y
+
+    def __repr__(self):
+        return "(%d,%d)" % (self.x, self.y)
+
+    def move(self, distance):
+        self.x += distance.x
+        self.y += distance.y
+
+
+class Size(object):
+    def __init__(self, width, height):
+        self.width = width
+        self.height = height
+
+    def __repr__(self):
+        return "%ux%u" % (self.width, self.height)
+
+
 class Rect(object):
     def __init__(self, left, top, width, height):
         self.left = left