diff --git a/src/scanner.c b/src/scanner.c index 57d6be0c2..c7fa5b10d 100644 --- a/src/scanner.c +++ b/src/scanner.c @@ -92,7 +92,7 @@ bool tree_sitter_latex_external_scanner_scan(void *payload, TSLexer *lexer, const bool *valid_symbols) { bool found = false; TSSymbol type = 0xFFFF; - for (int i = 0; i <= TRIVIA_RAW_ENV_MINTED; i++) { + for (int i = 0; i <= TRIVIA_RAW_ENV_PYCODE; i++) { if (valid_symbols[i]) { if (found) { return false; diff --git a/test/corpus/environments.txt b/test/corpus/environments.txt index 160cbdc95..4c787dd12 100644 --- a/test/corpus/environments.txt +++ b/test/corpus/environments.txt @@ -214,6 +214,28 @@ Minted environment (text (word)))))) +================================================================================ +Pycode environment +================================================================================ + +\begin{pycode} + print() +\end{pycode} + +-------------------------------------------------------------------------------- + +(source_file + (pycode_environment + (begin + (curly_group_text + (text + (word)))) + (source_code) + (end + (curly_group_text + (text + (word)))))) + ================================================================================ Math environment ================================================================================