BUG: Fix two argument size support for Fortran module routines. Reverted size-to-shape mapping patch and implemented two argument size function in C.

1 files changed, 16 insertions, 0 deletions

diff --git a/numpy/f2py/tests/test_size.py b/numpy/f2py/tests/test_size.py
index c00dd9a31..a548e9885 100644
--- a/numpy/f2py/tests/test_size.py
+++ b/numpy/f2py/tests/test_size.py
@@ -24,6 +24,22 @@ class TestSizeSumExample(util.F2PyTest):
         r = self.module.foo([[1,2],[3,4],[5,6]])
         assert_equal(r, [3,7,11],`r`)
 
+    @dec.slow
+    def test_transpose(self):
+        r = self.module.trans([[1,2]])
+        assert_equal(r, [[1],[2]],`r`)
+
+        r = self.module.trans([[1,2,3],[4,5,6]])
+        assert_equal(r, [[1,4],[2,5],[3,6]],`r`)
+
+    @dec.slow
+    def test_flatten(self):
+        r = self.module.flatten([[1,2]])
+        assert_equal(r, [1,2],`r`)
+
+        r = self.module.flatten([[1,2,3],[4,5,6]])
+        assert_equal(r, [1,2,3,4,5,6],`r`)
+
 if __name__ == "__main__":
     import nose
     nose.runmodule()