Simplify test.

ponder-lab · Dec 5, 2023 · b31a47b · b31a47b
1 parent 7c8c6cd
commit b31a47b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 27 deletions.
diff --git a/...n/testHasLikelyTensorParameter148/in/A.py → ...es/HybridizeFunction/testCallback/in/A.py b/...n/testHasLikelyTensorParameter148/in/A.py → ...es/HybridizeFunction/testCallback/in/A.py
diff --git a/...elyTensorParameter148/in/requirements.txt → ...Function/testCallback/in/requirements.txt b/...elyTensorParameter148/in/requirements.txt → ...Function/testCallback/in/requirements.txt
diff --git a/...ze.tests/test cases/edu/cuny/hunter/hybridize/tests/HybridizeFunctionRefactoringTest.java b/...ze.tests/test cases/edu/cuny/hunter/hybridize/tests/HybridizeFunctionRefactoringTest.java
@@ -4395,33 +4395,6 @@ public void testHasLikelyTensorParameter147() throws Exception {
 		testHasLikelyTensorParameterHelper(false, false);
 	}
 
-	/**
-	 * Test for https://github.com/ponder-lab/Hybridize-Functions-Refactoring/issues/280.
-	 */
-	@Test
-	public void testHasLikelyTensorParameter148() throws Exception {
-		Set<Function> functions = this.getFunctions();
-		assertNotNull(functions);
-		assertEquals(1, functions.size());
-		Function function = functions.iterator().next();
-		assertNotNull(function);
-		assertEquals(true, function.getIsHybrid());
-
-		argumentsType params = function.getParameters();
-
-		// one params.
-		exprType[] actualParams = params.args;
-		assertEquals(1, actualParams.length);
-
-		exprType actualParameter = actualParams[0];
-		assertNotNull(actualParameter);
-
-		String paramName = NodeUtils.getRepresentationString(actualParameter);
-		assertEquals("input", paramName);
-
-		assertTrue("Expecting function with likely tensor parameter.", function.getLikelyHasTensorParameter());
-	}
-
 	// TODO: Test arbitrary expression.
 	// TODO: Test cast/assert statements?
 	// TODO: https://www.tensorflow.org/guide/function#pass_tensors_instead_of_python_literals. How do we deal with union types? Do we want
@@ -5691,5 +5664,14 @@ public void testRecursion11() throws Exception {
 		assertTrue(f.getEntryMatchingFailure(IS_RECURSIVE).isError());
 	}
 
+	/**
+	 * Test for https://github.com/ponder-lab/Hybridize-Functions-Refactoring/issues/280.
+	 */
+	@Test
+	public void testCallback() throws Exception {
+		Function f = getFunction("replica_fn");
+		assertTrue(f.getLikelyHasTensorParameter());
+	}
+
 	// TODO: Left off at https://www.tensorflow.org/guide/function#depending_on_python_global_and_free_variables.
 }