matplotlib
diff --git a/‎lib/matplotlib/tests/test_pickle.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_pickle.py
+3Lines changed: 3 additions & 0 deletions b/‎lib/matplotlib/tests/test_pickle.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_pickle.py
+3Lines changed: 3 additions & 0 deletions
diff --git a/‎lib/matplotlib/transforms.py
Copy file name to clipboardExpand all lines: lib/matplotlib/transforms.py
+8-1Lines changed: 8 additions & 1 deletion b/‎lib/matplotlib/transforms.py
Copy file name to clipboardExpand all lines: lib/matplotlib/transforms.py
+8-1Lines changed: 8 additions & 1 deletion
@@ -279,6 +279,9 @@ def test_transform():
     assert_equal(obj.wrapper._child, obj.composite)
     # Check child -> parent links of TransformWrapper.
     assert_equal(list(obj.wrapper._parents.values()), [obj.composite2])
+    # Check input and output dimensions are set as expected.
+    assert_equal(obj.wrapper.input_dims, obj.composite.input_dims)
+    assert_equal(obj.wrapper.output_dims, obj.composite.output_dims)
 
 
 if __name__ == '__main__':
 
@@ -1552,17 +1552,24 @@ def __eq__(self, other):
         def __str__(self):
             return str(self._child)
 
+    # NOTE: Transform.__[gs]etstate__ should be sufficient when using only
+    # Python 3.4+.
     def __getstate__(self):
-        # only store the child and parents
+        # only store the child information and parents
         return {
             'child': self._child,
+            'input_dims': self.input_dims,
+            'output_dims': self.output_dims,
             # turn the weakkey dictionary into a normal dictionary
             'parents': dict(six.iteritems(self._parents))
         }
 
     def __setstate__(self, state):
         # re-initialise the TransformWrapper with the state's child
         self._init(state['child'])
+        # The child may not be unpickled yet, so restore its information.
+        self.input_dims = state['input_dims']
+        self.output_dims = state['output_dims']
         # turn the normal dictionary back into a WeakValueDictionary
         self._parents = WeakValueDictionary(state['parents'])