Save number cell from setUp function (#166)

* save number cell from setup + test and expected output

* Set Version: 0.1.56

* Update src/tests/integration/test-data/foundry/test/BlockParamsTest.t.sol

* update expected ouptut

---------

Co-authored-by: devops <[email protected]>

package/version

@@ -1 +1 @@
-0.1.55
+0.1.56

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "kontrol"
-version = "0.1.55"
+version = "0.1.56"
 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.55'
+VERSION: Final = '0.1.56'

src/kontrol/prove.py

@@ -371,6 +371,7 @@ def _method_to_cfg(
             )
         for final_node in final_states:
             new_accounts_cell = final_node.cterm.cell('ACCOUNTS_CELL')
+            number_cell = final_node.cterm.cell('NUMBER_CELL')
             new_accounts = [CTerm(account, []) for account in flatten_label('_AccountCellMap_', new_accounts_cell)]
             new_accounts_map = {account.cell('ACCTID_CELL'): account for account in new_accounts}
             test_contract_account = new_accounts_map[Foundry.address_TEST_CONTRACT()]
@@ -385,6 +386,7 @@ def _method_to_cfg(
             new_accounts_cell = KEVM.accounts([account.config for account in new_accounts_map.values()])
 
             new_init_cterm = CTerm(set_cell(init_cterm.config, 'ACCOUNTS_CELL', new_accounts_cell), [])
+            new_init_cterm = CTerm(set_cell(new_init_cterm.config, 'NUMBER_CELL', number_cell), [])
             new_node = cfg.create_node(new_init_cterm)
             cfg.create_edge(final_node.id, new_node.id, depth=1)
             new_node_ids.append(new_node.id)

src/tests/integration/test-data/foundry.k.expected

@@ -14,6 +14,7 @@ module FOUNDRY-MAIN
     imports public ScriptBase-VERIFICATION
     imports public TestBase-VERIFICATION
     imports public BlockParamsTest-VERIFICATION
+    imports public RollTest-VERIFICATION
     imports public BroadcastTest-VERIFICATION
     imports public Constants-VERIFICATION
     imports public Contract-VERIFICATION
@@ -183,6 +184,13 @@ module BlockParamsTest-VERIFICATION
 
 
 
+endmodule
+
+module RollTest-VERIFICATION
+    imports public RollTest-CONTRACT
+
+
+
 endmodule
 
 module BroadcastTest-VERIFICATION

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

@@ -36,3 +36,14 @@ contract BlockParamsTest is Test {
         assert(x >= 0);
     }
 }
+
+contract RollTest is Test {
+    function setUp() external {
+        vm.roll(123);
+    }
+
+    function test_roll_setup() external {
+        assert(block.number == 123);
+
+    }
+}