private void assertFormulaResult(Number result, Integer fo, Integer br, Integer bz) { Number[] arr = trans.transform(new MockInputRow().put(foo, fo).put(bar, br).put(baz, bz)); assertNotNull(arr); assertEquals(1, arr.length); if (result == null) { assertNull("Expected null but got: " + arr[0], arr[0]); } else { if (arr[0] == null) { fail("Expected " + result + " but got null!"); } double d1 = result.doubleValue(); double d2 = arr[0].doubleValue(); assertEquals(d1, d2); } }