Skip to content

Commit cd6694c

Browse files
committed
Add a model test.
1 parent 55698f5 commit cd6694c

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

edu.cuny.hunter.hybridize.tests/test cases/edu/cuny/hunter/hybridize/tests/HybridizeFunctionRefactoringTest.java

+14
Original file line numberDiff line numberDiff line change
@@ -1594,4 +1594,18 @@ public void testHasLikelyTensorParameter10() throws Exception {
15941594
// TODO: Test arbitrary expression.
15951595
// TODO: Test cast/assert statements?
15961596
// TODO: Test tf.Tensor-like things?
1597+
1598+
/**
1599+
* Test a model. No tf.function in this one.
1600+
*/
1601+
@Test
1602+
public void testModel() throws Exception {
1603+
Set<Function> functions = this.getFunctions();
1604+
assertNotNull(functions);
1605+
1606+
LOG.info("Found functions: " + functions.size());
1607+
1608+
// no hybrids.
1609+
assertTrue(functions.stream().map(Function::isHybrid).allMatch(b -> b == false));
1610+
}
15971611
}

0 commit comments

Comments
 (0)