Add support for strict generators

elixir-makeup · Jan 28, 2025 · c1206fb · c1206fb
1 parent 7ad4baa
commit c1206fb
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/lib/makeup/lexers/erlang_lexer.ex b/lib/makeup/lexers/erlang_lexer.ex
@@ -220,7 +220,7 @@ defmodule Makeup.Lexers.ErlangLexer do
 
   syntax_operators =
     word_from_list(
-      ~W[+ - +? ++ = == -- * / < > /= =:= =/= =< >= ==? <- <= ! ? ?!],
+      ~W[+ - +? ++ = == -- * / < > /= =:= =/= =< >= ==? <- <:- <= <:= ! ? ?!],
       :operator
     )
 

diff --git a/test/makeup/erlang_lexer/erlang_lexer_tokenizer_test.exs b/test/makeup/erlang_lexer/erlang_lexer_tokenizer_test.exs
@@ -281,6 +281,28 @@ defmodule ErlangLexerTokenizer do
              ]
     end
 
+    test "strict" do
+      assert lex("[A||A<:-B]") == [
+               {:punctuation, %{group_id: "group-1"}, "["},
+               {:name, %{}, "A"},
+               {:punctuation, %{}, "||"},
+               {:name, %{}, "A"},
+               {:operator, %{}, "<:-"},
+               {:name, %{}, "B"},
+               {:punctuation, %{group_id: "group-1"}, "]"}
+             ]
+
+      assert lex("[A||A<:=B]") == [
+               {:punctuation, %{group_id: "group-1"}, "["},
+               {:name, %{}, "A"},
+               {:punctuation, %{}, "||"},
+               {:name, %{}, "A"},
+               {:operator, %{}, "<:="},
+               {:name, %{}, "B"},
+               {:punctuation, %{group_id: "group-1"}, "]"}
+             ]
+    end
+
     test "parallel" do
       assert lex("[A||A<-B&&C<-D]") == [
                {:punctuation, %{group_id: "group-1"}, "["},
@@ -379,7 +401,9 @@ defmodule ErlangLexerTokenizer do
       assert lex("=") == [{:operator, %{}, "="}]
       assert lex("!") == [{:operator, %{}, "!"}]
       assert lex("<-") == [{:operator, %{}, "<-"}]
+      assert lex("<:-") == [{:operator, %{}, "<:-"}]
       assert lex("<=") == [{:operator, %{}, "<="}]
+      assert lex("<:=") == [{:operator, %{}, "<:="}]
     end
 
     test "word operators are tokenized as operator" do