Verify proveX and checkX functions similarly to testX

package/version

@@ -1 +1 @@
-0.1.107
+0.1.108

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "kontrol"
-version = "0.1.107"
+version = "0.1.108"
 description = "Foundry integration for KEVM"
 authors = [
     "Runtime Verification, Inc. <[email protected]>",

src/kontrol/__init__.py

@@ -5,4 +5,4 @@
 if TYPE_CHECKING:
     from typing import Final
 
-VERSION: Final = '0.1.107'
+VERSION: Final = '0.1.108'

src/kontrol/prove.py

@@ -413,7 +413,8 @@ def _method_to_cfg(
         new_node_ids = [init_node.id]
         init_node_id = init_node.id
 
-    is_test = method.signature.startswith('test')
+    proof_prefixes = ['test', 'proof', 'check']
+    is_test = any(method.signature.startswith(prefix) for prefix in proof_prefixes)
     failing = method.signature.startswith('testFail')
     final_cterm = _final_cterm(
         empty_config, contract.name_with_path, failing=failing, is_test=is_test, use_init_code=use_init_code

src/tests/integration/test-data/foundry-prove-all

@@ -38,6 +38,7 @@ ArithmeticTest.test_wmul_wdiv_inverse_underflow(uint256,uint256)
 ArithmeticTest.test_wmul_weakly_increasing_positive(uint256,uint256)
 AssertTest.test_assert_false()
 AssertTest.test_assert_true()
+AssertTest.prove_assert_true()
 AssertTest.test_assert_true_branch(uint256)
 AssertTest.testFail_assert_false()
 AssertTest.testFail_assert_true()

src/tests/integration/test-data/foundry-prove-skip-legacy

@@ -31,6 +31,7 @@ ArithmeticTest.test_wmul_rounding(uint256,uint256)
 ArithmeticTest.test_wmul_wdiv_inverse(uint256,uint256)
 ArithmeticTest.test_wmul_wdiv_inverse_underflow(uint256,uint256)
 ArithmeticTest.test_wmul_weakly_increasing_positive(uint256,uint256)
+AssertTest.prove_assert_true()
 AssertTest.test_assert_false()
 AssertTest.testFail_assert_true()
 AssertTest.testFail_expect_revert()

src/tests/integration/test-data/foundry/test/Simple.t.sol

@@ -21,6 +21,10 @@ contract AssertTest is Test, KEVMCheats {
         assert(true);
     }
 
+    function prove_assert_true() public pure {
+        assert(true);
+    }
+
     function test_assert_true_branch(uint x) public {
         if (x < 3) {
             y = x;