Added a Diagonal constraint.

[pseudoku.git] / pseudoku / __init__.py

diff --git a/pseudoku/__init__.py b/pseudoku/__init__.py
index 853904b..2fe15f3 100644 (file)
--- a/pseudoku/__init__.py
+++ b/pseudoku/__init__.py
@@ -1,4 +1,5 @@
 from grid import Grid
+from grid.cellgroup import Diagonal
 import render.text
 
 def main():
@@ -14,6 +15,9 @@ def main():
         1...6...9
     """)
 
+    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)