Added a Diagonal constraint.

[pseudoku.git] / pseudoku / __init__.py

diff --git a/pseudoku/__init__.py b/pseudoku/__init__.py
index 3b6eeb9..2fe15f3 100644 (file)
--- a/pseudoku/__init__.py
+++ b/pseudoku/__init__.py
@@ -1,4 +1,6 @@
 from grid import Grid
+from grid.cellgroup import Diagonal
+import render.text
 
 def main():
     grid = Grid.from_string("""
@@ -13,8 +15,12 @@ def main():
         1...6...9
     """)
 
-    print grid
+    grid.add_constraint(Diagonal(grid, direction='down', offset=0))
+    grid.add_constraint(Diagonal(grid, direction='up', offset=0))
+
+    r = render.text.AsciiArtGridRenderer()
+    print r.render_grid(grid)
 
     grid.solve()
 
-    print grid
+    print r.render_grid(grid)