arvids changes

Author: jack60612

chia/_tests/util/full_sync.py

@@ -142,7 +142,8 @@ async def run_sync_test(
         config = load_config(root_path, "config.yaml")
 
         if test_constants:
-            constants = TEST_CONSTANTS
+            # this allows all blocks have compressed generators
+            constants = TEST_CONSTANTS.replace(HARD_FORK_HEIGHT=0)
         else:
             overrides = config["network_overrides"]["constants"][config["selected_network"]]
             constants = replace_str_to_bytes(DEFAULT_CONSTANTS, **overrides)

chia/simulator/block_tools.py

@@ -403,7 +403,10 @@ def setup_new_gen(
             assert rng is not None
             bundle, additions = make_spend_bundle(available_coins, wallet, rng)
             removals = bundle.removals()
-            program = simple_solution_generator(bundle).program
+            if curr.height >= self.constants.HARD_FORK_HEIGHT:
+                program = simple_solution_generator_backrefs(bundle).program
+            else:
+                program = simple_solution_generator(bundle).program
             cost = compute_block_cost(program, self.constants, uint32(curr.height + 1), prev_tx_height)
             return NewBlockGenerator(
                 program,