Implement jupyter display for solutions.

polyomino/jupyter.py

@@ -0,0 +1,11 @@
+import io
+
+from pretty_poly.image import make_colored_blocks
+from pretty_poly.png import _make_png_writer
+
+def solution_to_png(solution):
+    data, palette = make_colored_blocks(solution.tiling)
+    writer = _make_png_writer(data, palette)
+    f = io.BytesIO()
+    writer.write(f, data)
+    return f.getvalue()

polyomino/solution.py

@@ -1,5 +1,7 @@
 from pretty_poly import make_ascii
 
+from .jupyter import solution_to_png
+
 
 class Solution(object):
     def __init__(self, tiling, board):
@@ -14,3 +16,6 @@ def python(self):
 
     def _gen_python(self):
         yield "TILING = %s" % (repr(self.tiling),)
+
+    def _repr_png_(self):
+        return solution_to_png(self)