Skip to content

Commit

Permalink
Revert conflict change with main branch
Browse files Browse the repository at this point in the history
Signed-off-by: zane-neo <[email protected]>
  • Loading branch information
zane-neo committed Jan 30, 2024
1 parent 248c670 commit dee7989
Showing 1 changed file with 5 additions and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -42,28 +42,16 @@ public void test_getResponseFilter() {
}

@Test
public void test_buildMultipleResultModelTensorList() {
Assert.assertNotNull(MLPostProcessFunction.buildModelTensorResult());
public void test_buildModelTensorList() {
Assert.assertNotNull(MLPostProcessFunction.buildModelTensorList());
List<List<Float>> numbersList = new ArrayList<>();
numbersList.add(Collections.singletonList(1.0f));
Assert.assertNotNull(MLPostProcessFunction.buildModelTensorResult().apply(numbersList));
Assert.assertNotNull(MLPostProcessFunction.buildModelTensorList().apply(numbersList));
}

@Test
public void test_buildMultipleResultModelTensorList_exception() {
public void test_buildModelTensorList_exception() {
exceptionRule.expect(IllegalArgumentException.class);
MLPostProcessFunction.buildModelTensorResult().apply(null);
}

@Test
public void test_buildSingleResultModelTensorList() {
Assert.assertNotNull(MLPostProcessFunction.buildSingleResultModelTensor());
Assert.assertNotNull(MLPostProcessFunction.buildSingleResultModelTensor().apply(Collections.singletonList(1.0f)));
}

@Test
public void test_buildSingleResultModelTensorList_exception() {
exceptionRule.expect(IllegalArgumentException.class);
MLPostProcessFunction.buildSingleResultModelTensor().apply(null);
MLPostProcessFunction.buildModelTensorList().apply(null);
}
}

0 comments on commit dee7989

Please sign in to comment.