Spaces:
Running
Running
Fix pattern border normalization
Browse files- tests/test_core.py +28 -0
tests/test_core.py
CHANGED
|
@@ -135,3 +135,31 @@ class CoreTests(unittest.TestCase):
|
|
| 135 |
)
|
| 136 |
self.assertIn("|..|##|", normalized)
|
| 137 |
self.assertIn("|##|..|", normalized)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 135 |
)
|
| 136 |
self.assertIn("|..|##|", normalized)
|
| 137 |
self.assertIn("|##|..|", normalized)
|
| 138 |
+
|
| 139 |
+
def test_pattern_normalization_keeps_top_and_bottom_borders(self) -> None:
|
| 140 |
+
board = "\n".join(
|
| 141 |
+
[
|
| 142 |
+
" 1 ",
|
| 143 |
+
" 1 2 3 3 3 ",
|
| 144 |
+
" +--+--+--+--+--+",
|
| 145 |
+
"2 1| | | | | |",
|
| 146 |
+
" +--+--+--+--+--+",
|
| 147 |
+
" 1| | | | | |",
|
| 148 |
+
" +--+--+--+--+--+",
|
| 149 |
+
" 3| | | | | |",
|
| 150 |
+
" +--+--+--+--+--+",
|
| 151 |
+
" 3| | | | | |",
|
| 152 |
+
" +--+--+--+--+--+",
|
| 153 |
+
" 3| | | | | |",
|
| 154 |
+
" +--+--+--+--+--+",
|
| 155 |
+
]
|
| 156 |
+
)
|
| 157 |
+
normalized = normalize_board_for_display(
|
| 158 |
+
puzzle_type="pattern",
|
| 159 |
+
problem_ascii=board,
|
| 160 |
+
board_ascii=board,
|
| 161 |
+
)
|
| 162 |
+
lines = normalized.splitlines()
|
| 163 |
+
self.assertEqual(lines.count(" +--+--+--+--+--+"), 6)
|
| 164 |
+
self.assertEqual(lines[2], " +--+--+--+--+--+")
|
| 165 |
+
self.assertEqual(lines[-1], " +--+--+--+--+--+")
|