Guess .pyi files as Python (#1996)

rouge-ruby · Sep 25, 2023 · b9d05fd · b9d05fd
1 parent b584ea1
commit b9d05fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/rouge/lexers/python.rb b/lib/rouge/lexers/python.rb
@@ -8,8 +8,8 @@ class Python < RegexLexer
       desc "The Python programming language (python.org)"
       tag 'python'
       aliases 'py'
-      filenames '*.py', '*.pyw', '*.sc', 'SConstruct', 'SConscript', '*.tac',
-                '*.bzl', 'BUCK', 'BUILD', 'BUILD.bazel', 'WORKSPACE'
+      filenames '*.py', '*.pyi', '*.pyw', '*.sc', 'SConstruct', 'SConscript',
+                '*.tac', '*.bzl', 'BUCK', 'BUILD', 'BUILD.bazel', 'WORKSPACE'
       mimetypes 'text/x-python', 'application/x-python'
 
       def self.detect?(text)

diff --git a/spec/lexers/python_spec.rb b/spec/lexers/python_spec.rb
@@ -9,6 +9,7 @@
 
     it 'guesses by filename' do
       assert_guess :filename => 'foo.py'
+      assert_guess :filename => 'foo.pyi'
       assert_guess :filename => 'foo.pyw'
       assert_guess :filename => '*.sc', :source => '# A comment'
       assert_guess :filename => 'SConstruct'