free: fix one test

aya-prover · Jan 4, 2025 · fc0f4ae · fc0f4ae
1 parent f2787de
commit fc0f4ae
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/jit-compiler/src/main/java/org/aya/compiler/free/morphism/free/FreeArgumentProvider.java b/jit-compiler/src/main/java/org/aya/compiler/free/morphism/free/FreeArgumentProvider.java
@@ -1,4 +1,4 @@
-// Copyright (c) 2020-2024 Tesla (Yinsen) Zhang.
+// Copyright (c) 2020-2025 Tesla (Yinsen) Zhang.
 // Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
 package org.aya.compiler.free.morphism.free;
 
@@ -8,22 +8,19 @@
 import org.jetbrains.annotations.NotNull;
 
 public record FreeArgumentProvider(int paramCount) implements ArgumentProvider {
-  @Override
-  public @NotNull LocalVariable arg(int nth) {
+  @Override public @NotNull LocalVariable arg(int nth) {
     assert nth < paramCount;
     return new FreeVariable.Arg(nth);
   }
 
   record Lambda(int captureCount, int paramCount) implements ArgumentProvider.Lambda {
-    @Override
-    public @NotNull FreeJavaExpr capture(int nth) {
+    @Override public @NotNull FreeJavaExpr capture(int nth) {
       assert nth < captureCount;
       return new FreeExpr.RefCapture(nth);
     }
 
-    @Override
-    public @NotNull LocalVariable arg(int nth) {
-      assert nth + captureCount < paramCount;
+    @Override public @NotNull LocalVariable arg(int nth) {
+      assert nth < paramCount;
       return new FreeVariable.Arg(nth);
     }
   }