nilshoehing commited on
Commit
52f5c59
·
verified ·
1 Parent(s): fcda80c

Fix pattern border normalization

Browse files
Files changed (1) hide show
  1. 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], " +--+--+--+--+--+")