starknet_os_flow_tests: implement get_builtin_usage

Author: dorimedini-starkware

crates/starknet_os_flow_tests/src/test_manager.rs

@@ -13,6 +13,7 @@ use blockifier::transaction::objects::TransactionExecutionInfo;
 use blockifier::transaction::transaction_execution::Transaction as BlockifierTransaction;
 use blockifier_test_utils::calldata::create_calldata;
 use blockifier_test_utils::contracts::FeatureContract;
+use cairo_vm::types::builtin_name::BuiltinName;
 use itertools::Itertools;
 use starknet_api::abi::abi_utils::get_fee_token_var_address;
 use starknet_api::block::{BlockHash, BlockInfo, BlockNumber, PreviousBlockNumber};
@@ -130,6 +131,16 @@ pub(crate) struct OsTestOutput<S: FlowTestState> {
 }
 
 impl<S: FlowTestState> OsTestOutput<S> {
+    pub(crate) fn get_builtin_usage(&self, builtin_name: &BuiltinName) -> usize {
+        *self
+            .runner_output
+            .metrics
+            .execution_resources
+            .builtin_instance_counter
+            .get(builtin_name)
+            .unwrap()
+    }
+
     pub(crate) fn perform_default_validations(&self) {
         self.perform_validations(true, None);
     }

crates/starknet_os_flow_tests/src/tests.rs

@@ -1515,22 +1515,16 @@ async fn test_new_class_flow(#[case] use_kzg_da: bool, #[case] n_blocks_in_multi
 
     // Validate poseidon usage.
     // TODO(Meshi): Add blake opcode validations.
-    let poseidons = test_output
-        .runner_output
-        .metrics
-        .execution_resources
-        .builtin_instance_counter
-        .get(&BuiltinName::poseidon)
-        .unwrap();
+    let poseidons = test_output.get_builtin_usage(&BuiltinName::poseidon);
     if use_kzg_da {
         expect![[r#"
             679
         "#]]
-        .assert_debug_eq(poseidons);
+        .assert_debug_eq(&poseidons);
     } else {
         expect![[r#"
             562
         "#]]
-        .assert_debug_eq(poseidons);
+        .assert_debug_eq(&poseidons);
     }
 }