Fixed one of the tests

Author: sumagnadas

modules/geometry.py

@@ -1,5 +1,6 @@
 from PySide2.QtCore import QPoint, Qt
 from PySide2 import QtGui
+import math
 
 
 class Point(QPoint):
@@ -41,3 +42,15 @@ def plot(window, closedflag):
                               window.pointNum,
                               closedflag)
     window.image.setPixmap(QtGui.QPixmap().fromImage(graph))
+
+def dist(a, b):
+    x0 = int(a.x())
+    y0 = int(a.y())
+    x3 = int(b.x())
+    y3 = int(b.y())
+    x1 = x0 if x0 >= x3 else x3
+    y1 = y0 if y0 >= y3 else y3
+    x2 = x0 if x0 < x3 else x3
+    y2 = y0 if y0 < y3 else y3
+    dist = math.sqrt(((x1 - x2) ** 2) + ((y1 - y2) ** 2))
+    return dist

tests/test_button.py

@@ -1,6 +1,6 @@
 import pytestqt
-from modules.gui import QtGui
-from modules.gui import Graph, QtWidgets
+from modules.gui import Graph, QtWidgets, QtGui
+from modules.geometry import dist, Point
 from PySide2.QtCore import Qt
 from os import path
 
@@ -16,11 +16,13 @@ def test_draw(qtbot):
     qtbot.waitForWindowShown(window)
 
     qtbot.mouseClick(window.buttonBox.buttons()[0], Qt.LeftButton)
-
-    assert window.image.grab().toImage().pixelColor(window.p1) == Qt.black, ('Line not drawn'
-                                                             'or button '
-                                                             'not working')
-    assert window.image.grab().toImage().pixelColor(window.p2) == Qt.black, ('Line not drawn'
+    graph = window.image.grab().toImage()
+    
+    for i in range(-20, 20):
+        for j in range(-20, 20):
+            p = Point(i, j)
+            if dist(window.p1, p) + dist(window.p2, p) == dist(window.p1, window.p2):
+                assert graph.pixelColor(p) == Qt.black, ('Line not drawn'
                                                              'or button '
                                                              'not working')