Add SquareSet.from_square().

Author: niklasf

CHANGELOG.rst

@@ -27,6 +27,8 @@ Minor new features:
 
 * Allow `chess.Board.set_piece_at(square, None)`.
 
+* Added `chess.SquareSet.from_square(square)`.
+
 New in v0.13.0
 --------------
 

chess/__init__.py

@@ -4097,3 +4097,13 @@ def __str__(self):
 
     def __hash__(self):
         return hash(self.mask)
+
+    @classmethod
+    def from_square(cls, square):
+        """
+        Creates a SquareSet from a single square.
+
+        >>> chess.SquareSet.from_square(chess.A1) == chess.BB_A1:
+        True
+        """
+        return cls(BB_SQUARES[square])

test.py

@@ -1253,6 +1253,10 @@ def test_mutable_set_operations(self):
         self.assertEqual(squares.pop(), chess.C7)
         self.assertEqual(squares, chess.BB_VOID)
 
+    def test_from_square(self):
+        self.assertEqual(chess.SquareSet.from_square(chess.H5), chess.BB_H5)
+        self.assertEqual(chess.SquareSet.from_square(chess.C2), chess.BB_C2)
+
 
 class PolyglotTestCase(unittest.TestCase):