From 181bc7d61e79949ea1ca2e063b6d9053babcc8b4 Mon Sep 17 00:00:00 2001 From: Eryn Wells Date: Wed, 11 Oct 2017 19:43:15 -0700 Subject: [PATCH] Remove idx from peer set --- sudoku/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sudoku/__init__.py b/sudoku/__init__.py index fd29b5f..cbe5b38 100644 --- a/sudoku/__init__.py +++ b/sudoku/__init__.py @@ -110,8 +110,9 @@ class Sudoku: ''' Return a set of the peers, indexes into the board, for a given square. ''' + idx = self._xy_to_idx(x, y) box = int(x / self.size), int(y / self.size) - return set(self._row(y)) | set(self._column(x)) | set(self._box(*box)) + return (set(self._row(y)) | set(self._column(x)) | set(self._box(*box))) - {idx} def _row(self, r): row_size = self.row_size