From 529f6da8a1fa76baf4e0d7e8d332b5ead56b48de Mon Sep 17 00:00:00 2001
From: Georg Brandl <georg@python.org>
Date: Mon, 20 May 2013 07:42:12 +0200
Subject: Closes #863: improve analyse_text of Python and Numpy lexers

---
 pygments/lexers/agile.py | 3 ++-
 pygments/lexers/math.py  | 5 +++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/pygments/lexers/agile.py b/pygments/lexers/agile.py
index 896a3deb..1f81365e 100644
--- a/pygments/lexers/agile.py
+++ b/pygments/lexers/agile.py
@@ -185,7 +185,8 @@ class PythonLexer(RegexLexer):
     }
 
     def analyse_text(text):
-        return shebang_matches(text, r'pythonw?(2(\.\d)?)?')
+        return shebang_matches(text, r'pythonw?(2(\.\d)?)?') or \
+            'import ' in text[:1000]
 
 
 class Python3Lexer(RegexLexer):
diff --git a/pygments/lexers/math.py b/pygments/lexers/math.py
index a72a9124..f0e49fef 100644
--- a/pygments/lexers/math.py
+++ b/pygments/lexers/math.py
@@ -983,6 +983,11 @@ class NumPyLexer(PythonLexer):
             else:
                 yield index, token, value
 
+    def analyse_text(text):
+        return (shebang_matches(text, r'pythonw?(2(\.\d)?)?') or
+                'import ' in text[:1000]) \
+            and ('import numpy' in text or 'from numpy import' in text)
+
 
 class RConsoleLexer(Lexer):
     """
-- 
cgit v1.2.1