fix on context entities

subgraph/schema.graphql

@@ -363,7 +363,7 @@ type Transaction @entity(immutable: true) {
   events: [Event!] @derivedFrom(field: "transaction")
 }
 
-type ContextEntity implements Event @entity {
+type ContextEntity @entity {
   id: ID!
   "Base caller"
   caller: Account!

subgraph/src/utils.ts

@@ -376,9 +376,7 @@ export function createSignedContext(
   }
   return new SignedContext("");
 }
-export function createContextEntity(
-  txHash: string
-): ContextEntity {
+export function createContextEntity(txHash: string): ContextEntity {
   for (let i = 0; ; i++) {
     let contextEntity = ContextEntity.load(`${txHash}-${i}`);
     if (!contextEntity) {

subgraph/tests/entities.rs

@@ -2847,12 +2847,6 @@ async fn take_order_entity_take_order_test() -> anyhow::Result<()> {
     )
     .await?;
 
-    let vault_balance_alice: U256 = orderbook
-        .vault_balance(alice.address(), token_output.address(), vault_id)
-        .call()
-        .await?;
-    println!("vault_balance_alice_0: {}", vault_balance_alice);
-
     // Take the order
     let take_order_func = orderbook
         .connect(&bob)
@@ -2867,15 +2861,8 @@ async fn take_order_entity_take_order_test() -> anyhow::Result<()> {
 
     let block_data = get_block_data(&take_order_tx_hash).await?;
 
-    let vault_balance_alice: U256 = orderbook
-        .vault_balance(alice.address(), token_output.address(), vault_id)
-        .call()
-        .await?;
-    println!("vault_balance_alice_1: {}", vault_balance_alice);
-
     // Using index 0 since only one take order was made in this tx
     let take_order_entity = format!("{:?}-{}", take_order_tx_hash, 0);
-    println!("take_order_entity: {}", take_order_entity);
 
     let input_token = take_order_event
         .config

subgraph/tests/utils/numbers.rs

@@ -38,6 +38,7 @@ pub fn display_number(number: U256, decimals: u8) -> String {
     result
 }
 
+/// TODO: WOrk in return a Result<Option<String>>
 pub fn divide_decimal_strings(value_1: &str, value_2: &str) -> Option<String> {
     // Parse the input strings into BigDecimal values
     let decimal_1 = BigDecimal::from_str(value_1).expect("Invalid decimal value for value_1");
@@ -51,8 +52,10 @@ pub fn divide_decimal_strings(value_1: &str, value_2: &str) -> Option<String> {
     // Perform the division
     let result = decimal_1 / decimal_2;
 
+    let formatted_result = format!("{:.2}", result);
+
     // Format the result as a string with a specific precision
-    let formatted_result = format!("{:.1}", result); // Adjust the precision as needed
+    let formatted_value = formatted_result.trim_end_matches('0').trim_end_matches('.');
 
-    Some(formatted_result)
+    Some(formatted_value.to_string())
 }