ethereum_test_tools/spec: fix difficulty in genesis

Author: marioevz

src/ethereum_test_tools/spec/blockchain_test.py

@@ -67,7 +67,7 @@ def make_genesis(
             transactions_root=EmptyTrieRoot,
             receipt_root=EmptyTrieRoot,
             bloom="0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",  # noqa: E501
-            difficulty=0x20000,
+            difficulty=0x20000 if env.difficulty is None else env.difficulty,
             number=0,
             gas_limit=env.gas_limit,
             gas_used=0,

src/ethereum_test_tools/spec/state_test.py

@@ -66,7 +66,7 @@ def make_genesis(
             transactions_root=EmptyTrieRoot,
             receipt_root=EmptyTrieRoot,
             bloom="0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",  # noqa: E501
-            difficulty=0x20000,
+            difficulty=0x20000 if env.difficulty is None else env.difficulty,
             number=env.number - 1,
             gas_limit=env.gas_limit,
             # We need the base fee to remain unchanged from the genesis