gas_diff_stats.py: Include code deposit costs in total cost

Author: cameel

scripts/gas_diff_stats.py

@@ -23,7 +23,7 @@
 import sys
 from pathlib import Path
 from enum import Enum
-from parsec import generate, ParseError, regex, string
+from parsec import generate, ParseError, regex, string, optional
 from tabulate import tabulate
 
 class Kind(Enum):
@@ -49,6 +49,7 @@ class Diff(Enum):
 gas_ir_optimized = string("gas irOptimized").result(Kind.IrOptimized)
 gas_legacy_optimized = string("gas legacyOptimized").result(Kind.LegacyOptimized)
 gas_legacy = string("gas legacy").result(Kind.Legacy)
+code_suffix = string("code")
 
 def number() -> int:
     """Parse number."""
@@ -67,6 +68,8 @@ def diff_string() -> (Kind, Diff, int):
     yield comment
     yield space
     codegen_kind = yield gas_ir_optimized ^ gas_ir ^ gas_legacy_optimized ^ gas_legacy
+    yield optional(space)
+    yield optional(code_suffix)
     yield colon
     yield space
     val = yield number()

test/scripts/test_gas_diff_stats.py

@@ -85,3 +85,25 @@ def test_collect_statistics_should_ignore_unchanged_costs(self):
             0, # +legacyOptimized
             3, # +legacy
         ))
+
+    def test_collect_statistics_should_include_code_deposit_in_total_cost(self):
+        diff_output = dedent("""
+            -// gas irOptimized: 1
+            -// gas legacy: 20
+            -// gas legacyOptimized: 300
+            +// gas irOptimized: 4000
+            +// gas irOptimized code: 50000
+            +// gas legacy: 600000
+            +// gas legacyOptimized: 7000000
+            +// gas legacyOptimized code: 80000000
+            -// gas legacy code: 900000000
+        """).splitlines()
+
+        self.assertEqual(collect_statistics(diff_output), (
+            1,         # -irOptimized
+            300,       # -legacyOptimized
+            900000020, # -legacy
+            54000,     # +irOptimized
+            87000000,  # +legacyOptimized
+            600000,    # +legacy
+        ))