feat(julia): add tests for new changes

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/query/highlights/julia/test.jl b/tests/query/highlights/julia/test.jl
new file mode 100644
index 000000000..46fc313c0
--- /dev/null
+++ b/tests/query/highlights/julia/test.jl
@@ -0,0 +1,26 @@
+function load_data(::Symbol; ::Int) :: Tuple
+# <- keyword.function
+#        ^ function
+#                 ^ punctuation.bracket
+#                  ^^ punctuation.delimiter
+#                    ^ type.builtin
+#                          ^ punctuation.delimiter
+#                            ^^ punctuation.delimiter
+#                              ^^^ type.builtin
+#                                 ^ punctuation.bracket
+#                                   ^^ punctuation.delimiter
+#                                      ^ type.builtin
+    dataset = CIFAR10(; Tx = Float32, split = split)
+#   ^^^^^^^ variable
+#           ^ operator
+#             ^ function.call
+#                          ^ operator
+#                            ^ type.builtin
+    X = reshape(dataset.features[:, :, :, begin:n_obs], :, n_obs) # flattening the image pixels
+#                                         ^^^^^ variable.builtin
+    y = categorical2onehot(dataset.targets[begin:n_obs], N_LABELS)
+#                                          ^^^^^ variable.builtin
+    return X, y
+#   ^^^^^^ keyword.return
+end
+# <- keyword.function