From b6b875e930b93051604bea157950f0c639a14846 Mon Sep 17 00:00:00 2001
From: tiagobstr <tiago.baptista@strumenta.com>
Date: Tue, 11 Feb 2025 15:13:58 +0000
Subject: [PATCH] applying code review suggestions

---
 .../com/strumenta/kolasu/transformation/ASTTransformerTest.kt | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/core/src/test/kotlin/com/strumenta/kolasu/transformation/ASTTransformerTest.kt b/core/src/test/kotlin/com/strumenta/kolasu/transformation/ASTTransformerTest.kt
index 09cb8e89..e3856b39 100644
--- a/core/src/test/kotlin/com/strumenta/kolasu/transformation/ASTTransformerTest.kt
+++ b/core/src/test/kotlin/com/strumenta/kolasu/transformation/ASTTransformerTest.kt
@@ -536,6 +536,10 @@ class ASTTransformerTest {
             transformedAST.walkDescendants(AC::class).first().origin,
             original.walkDescendants(AC::class).first()
         )
+        assertEquals(
+            transformedAST.walkDescendants(AD::class).first().origin,
+            original.walkDescendants(AD::class).first()
+        )
     }
 }
 data class BazRoot(var stmts: MutableList<BazStmt> = mutableListOf()) : Node()