misc: more fixes ig?

aya-prover · Nov 23, 2023 · db52755 · db52755
1 parent 6d5ab52
commit db52755
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/base/src/test/java/org/aya/concrete/ParseTest.java b/base/src/test/java/org/aya/concrete/ParseTest.java
@@ -24,8 +24,7 @@
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.*;
 
 @SuppressWarnings("UnknownLanguage")
 public class ParseTest {
@@ -176,15 +175,15 @@ private void parseImport(@Language("Aya") String code) {
   }
 
   private void parseOpen(@Language("Aya") String code) {
-    assertTrue(parseStmt(code).last() instanceof Command.Open s && !s.toDoc(AyaPrettierOptions.debug()).debugRender().isEmpty());
+    assertTrue(parseStmt(code).getLast() instanceof Command.Open s && !s.toDoc(AyaPrettierOptions.debug()).debugRender().isEmpty());
   }
 
   private void parseFn(@Language("Aya") String code) {
     assertTrue(parseDecl(code).getFirst() instanceof TeleDecl.FnDecl s && !s.toDoc(AyaPrettierOptions.debug()).debugRender().isEmpty());
   }
 
   private void parseData(@Language("Aya") String code) {
-    assertTrue(parseDecl(code).getFirst() instanceof TeleDecl.DataDecl);
+    assertInstanceOf(TeleDecl.DataDecl.class, parseDecl(code).getFirst());
   }
 
   @Test