@Test
public void buildWithStagedThrows() {

  Throwable thrown =
      assertThrows(
          IllegalStateException.class,
          () -> ArgumentList.builder().stageFunction("col1", AggregateFunctions.MAX).build());
  assertTrue(thrown.getMessage().contains("Cannot build when a column is staged"));
}