Add parse_flags function + some unit tests.

1 files changed, 23 insertions, 1 deletions

diff --git a/numpy/distutils/tests/test_npy_pkg_config.py b/numpy/distutils/tests/test_npy_pkg_config.py
index cdf3d6902..c1de3ae23 100644
--- a/numpy/distutils/tests/test_npy_pkg_config.py
+++ b/numpy/distutils/tests/test_npy_pkg_config.py
@@ -2,7 +2,7 @@ import os
 from tempfile import mkstemp
 
 from numpy.testing import *
-from numpy.distutils.npy_pkg_config import read_config
+from numpy.distutils.npy_pkg_config import read_config, parse_flags
 
 simple = """\
 [meta]
@@ -70,3 +70,25 @@ class TestLibraryInfo(TestCase):
             self.failUnless(out.cflags() == '-I/Users/david/include')
         finally:
             os.remove(filename)
+
+class TestParseFlags(TestCase):
+    def test_simple_cflags(self):
+        d = parse_flags("-I/usr/include")
+        self.failUnless(d['include_dirs'] == ['/usr/include'])
+
+        d = parse_flags("-I/usr/include -DFOO")
+        self.failUnless(d['include_dirs'] == ['/usr/include'])
+        self.failUnless(d['macros'] == ['FOO'])
+
+        d = parse_flags("-I /usr/include -DFOO")
+        self.failUnless(d['include_dirs'] == ['/usr/include'])
+        self.failUnless(d['macros'] == ['FOO'])
+
+    def test_simple_lflags(self):
+        d = parse_flags("-L/usr/lib -lfoo -L/usr/lib -lbar")
+        self.failUnless(d['library_dirs'] == ['/usr/lib', '/usr/lib'])
+        self.failUnless(d['libs'] == ['foo', 'bar'])
+
+        d = parse_flags("-L /usr/lib -lfoo -L/usr/lib -lbar")
+        self.failUnless(d['library_dirs'] == ['/usr/lib', '/usr/lib'])
+        self.failUnless(d['libs'] == ['foo', 'bar'])