From b8eccd0f51e6ce247b0323db5222d9a68c955137 Mon Sep 17 00:00:00 2001 From: Raffi Khatchadourian Date: Wed, 24 Jul 2024 18:21:34 -0400 Subject: [PATCH] Be a little more lenient on what constitutes a test case entrypoint (#215) Also allow "container" (e,g., classes) names to end with "Test." There are other testing framework derivatives in Python that allow for this, apparently. --- .../cast/python/ipa/callgraph/PytestEntrypointBuilder.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/com.ibm.wala.cast.python/source/com/ibm/wala/cast/python/ipa/callgraph/PytestEntrypointBuilder.java b/com.ibm.wala.cast.python/source/com/ibm/wala/cast/python/ipa/callgraph/PytestEntrypointBuilder.java index cb3ab24a3..dac56b80f 100644 --- a/com.ibm.wala.cast.python/source/com/ibm/wala/cast/python/ipa/callgraph/PytestEntrypointBuilder.java +++ b/com.ibm.wala.cast.python/source/com/ibm/wala/cast/python/ipa/callgraph/PytestEntrypointBuilder.java @@ -82,7 +82,8 @@ public static boolean isPytestCase(IClass klass) { IClass container = dmb.getContainer(); String containerName = container.getReference().getName().getClassName().toString(); - if (containerName.startsWith("Test") && container instanceof PythonClass) { + if ((containerName.startsWith("Test") || containerName.endsWith("Test")) + && container instanceof PythonClass) { // It's a test class. PythonClass containerClass = (PythonClass) container;