improved set selection algorithm

1 files changed, 8 insertions, 0 deletions

diff --git a/window.py b/window.py
index 8f34683..fbf19be 100644
--- a/window.py
+++ b/window.py
@@ -217,6 +217,14 @@ def set_check(cardarray):
       cardarray[1].color != cardarray[2].color and \
       cardarray[0].color == cardarray[2].color:
       return False
+   if cardarray[0].color != cardarray[1].color and \
+      cardarray[1].color == cardarray[2].color and \
+      cardarray[0].color != cardarray[2].color:
+      return False
+   if cardarray[0].color == cardarray[1].color and \
+      cardarray[1].color == cardarray[2].color and \
+      cardarray[0].color != cardarray[2].color:
+      return False
    return True
    """
    if (cardarray[0].num == cardarray[1].num):