Merge pull request #7 from alcarney/develop

New Release v0.4.2
swyddfa · Sep 17, 2018 · 1a37948 · 1a37948
2 parents f3a2f89 + b0ab4db
commit 1a37948
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 3 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,3 +1,13 @@
+v0.4.2 - 2018-09-17
+-------------------
+
+Added
+^^^^^
+
+- :code:`Image` objects can now take a :code:`size` keyword argument to adjust
+  the size of the matplotlib preview plots
+
+
 v0.4.1 - 2018-09-17
 -------------------
 

diff --git a/stylo/_version.py b/stylo/_version.py
@@ -1 +1 @@
-__version__ = "0.4.1"
+__version__ = "0.4.2"
diff --git a/stylo/image/simple.py b/stylo/image/simple.py
@@ -5,15 +5,21 @@
 
 
 class Image(ABC):
-    def __call__(self, width, height, filename=None):
+    def __call__(self, width, height, filename=None, size=None):
 
         image = self._render(width, height)
 
         if filename is not None:
             self._save(image, filename)
             return
 
-        return plt.imshow(image)
+        if size is None:
+            size = (4, 4)
+
+        fig, ax = plt.subplots(1, figsize=size)
+        ax.imshow(image)
+
+        return fig
 
     def _save(self, image, filename):