Added test for new OSX nvcc args to be removed

theano/sandbox/cuda/tests/test_nvcc_compiler.py

@@ -25,16 +25,26 @@ def test_remove_python_framework_dir():
     # as otherwise it is difficult to do it properly.
     try:
         warnings.catch_warnings
+        test_warning = True
     except AttributeError:
-        return
-    cmd.append('Frameworks/Python.framework/Versions/2.6/Python')
-    # Python 2.4 "emulation" of `with` statement. It is necessary even if this
-    # code is not executed, because using `with` would result in a SyntaxError.
-    with_context = warnings.catch_warnings(record=True)
-    record = with_context.__enter__()
-    try:
-        assert remove_python_framework_dir(cmd) == cmd[0:-3] + cmd[-2:-1]
-        assert len(record) == 1
-        assert 'remove_python_framework_dir' in str(record[0].message)
-    finally:
-        with_context.__exit__(None, None, None)
+        test_warning = False
+    if test_warning:
+        cmd.append('Frameworks/Python.framework/Versions/2.6/Python')
+        # Python 2.4 "emulation" of `with` statement. It is necessary even if this
+        # code is not executed, because using `with` would result in a SyntaxError.
+        with_context = warnings.catch_warnings(record=True)
+        record = with_context.__enter__()
+        try:
+            assert remove_python_framework_dir(cmd) == cmd[0:-3] + cmd[-2:-1]
+            assert len(record) == 1
+            assert 'remove_python_framework_dir' in str(record[0].message)
+        finally:
+            with_context.__exit__(None, None, None)
+
+    # Now test some more typical arguments that should be caught by the regex.
+    for arg_to_remove in [
+            '/Library/Frameworks/EPD64.framework/Versions/7.1/Python',
+            '/Library/Frameworks/Python.framework/Versions/7.2/Python',
+            ]:
+        # Make sure those arguments are removed.
+        assert not remove_python_framework_dir([arg_to_remove])