From 985fb9706bce591c01e1acc7f71291e34eda41b7 Mon Sep 17 00:00:00 2001
From: Pedram_Mohajer <48964282+pedram-mohajer@users.noreply.github.com>
Date: Sun, 26 Nov 2023 21:40:24 -0500
Subject: [PATCH] Add doctest to is_safe function

---
 backtracking/n_queens.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/backtracking/n_queens.py b/backtracking/n_queens.py
index 0f237d95e7c8..2cd8c703fc72 100644
--- a/backtracking/n_queens.py
+++ b/backtracking/n_queens.py
@@ -24,6 +24,10 @@ def is_safe(board: list[list[int]], row: int, column: int) -> bool:
     Returns:
     Boolean Value
 
+    >>> is_safe([[0, 0, 0], [0, 0, 0], [0, 0, 0]], 1, 1)
+    True
+    >>> is_safe([[1, 0, 0], [0, 0, 0], [0, 0, 0]], 1, 1)
+    False
     """
 
     n = len(board)  # Size of the board