diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/Domain.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/Domain.kt
index 1a8a296c56..40e751b6c0 100644
--- a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/Domain.kt
+++ b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/Domain.kt
@@ -73,6 +73,7 @@ const val JUNIT5_PARAMETERIZED_PACKAGE = "org.junit.jupiter.params"
 
 //JUnit5 imports
 private const val TEST_NG_ASSERTIONS = "org.testng.Assert"
+private const val TEST_NG_ARRAYS_ASSERTIONS = "org.testng.internal.junit.ArrayAsserts"
 private const val JUNIT5_ASSERTIONS = "org.junit.jupiter.api.Assertions"
 private const val JUNIT4_ASSERTIONS = "org.junit.Assert"
 
@@ -170,6 +171,7 @@ sealed class TestFramework(
     var isInstalled: Boolean = false
     abstract val mainPackage: String
     abstract val assertionsClass: ClassId
+    abstract val arraysAssertionsClass: ClassId
     abstract val testAnnotation: String
     abstract val testAnnotationId: ClassId
     abstract val testAnnotationFqn: String
@@ -186,23 +188,23 @@ sealed class TestFramework(
 
     val assertDoubleEquals by lazy { assertionId("assertEquals", doubleClassId, doubleClassId, doubleClassId) }
 
-    val assertArrayEquals by lazy { assertionId("assertArrayEquals", Array<Any>::class.id, Array<Any>::class.id) }
+    val assertArrayEquals by lazy { arrayAssertionId("assertArrayEquals", Array<Any>::class.id, Array<Any>::class.id) }
 
-    val assertBooleanArrayEquals by lazy { assertionId("assertArrayEquals", booleanArrayClassId, booleanArrayClassId) }
+    open val assertBooleanArrayEquals by lazy { assertionId("assertArrayEquals", booleanArrayClassId, booleanArrayClassId) }
 
-    val assertByteArrayEquals by lazy { assertionId("assertArrayEquals", byteArrayClassId, byteArrayClassId) }
+    val assertByteArrayEquals by lazy { arrayAssertionId("assertArrayEquals", byteArrayClassId, byteArrayClassId) }
 
-    val assertCharArrayEquals by lazy { assertionId("assertArrayEquals", charArrayClassId, charArrayClassId) }
+    val assertCharArrayEquals by lazy { arrayAssertionId("assertArrayEquals", charArrayClassId, charArrayClassId) }
 
-    val assertShortArrayEquals by lazy { assertionId("assertArrayEquals", shortArrayClassId, shortArrayClassId) }
+    val assertShortArrayEquals by lazy { arrayAssertionId("assertArrayEquals", shortArrayClassId, shortArrayClassId) }
 
-    val assertIntArrayEquals by lazy { assertionId("assertArrayEquals", intArrayClassId, intArrayClassId) }
+    val assertIntArrayEquals by lazy { arrayAssertionId("assertArrayEquals", intArrayClassId, intArrayClassId) }
 
-    val assertLongArrayEquals by lazy { assertionId("assertArrayEquals", longArrayClassId, longArrayClassId) }
+    val assertLongArrayEquals by lazy { arrayAssertionId("assertArrayEquals", longArrayClassId, longArrayClassId) }
 
-    val assertFloatArrayEquals by lazy { assertionId("assertArrayEquals", floatArrayClassId, floatArrayClassId, floatClassId) }
+    val assertFloatArrayEquals by lazy { arrayAssertionId("assertArrayEquals", floatArrayClassId, floatArrayClassId, floatClassId) }
 
-    val assertDoubleArrayEquals by lazy { assertionId("assertArrayEquals", doubleArrayClassId, doubleArrayClassId, doubleClassId) }
+    val assertDoubleArrayEquals by lazy { arrayAssertionId("assertArrayEquals", doubleArrayClassId, doubleArrayClassId, doubleClassId) }
 
     val assertNull by lazy { assertionId("assertNull", objectClassId) }
 
@@ -212,8 +214,10 @@ sealed class TestFramework(
 
     val assertNotEquals by lazy { assertionId("assertNotEquals", objectClassId, objectClassId) }
 
-    private fun assertionId(name: String, vararg params: ClassId): MethodId =
+    protected fun assertionId(name: String, vararg params: ClassId): MethodId =
         builtinStaticMethodId(assertionsClass, name, voidClassId, *params)
+    private fun arrayAssertionId(name: String, vararg params: ClassId): MethodId =
+            builtinStaticMethodId(arraysAssertionsClass, name, voidClassId, *params)
 
     abstract fun getRunTestsCommand(
         executionInvoke: String,
@@ -251,6 +255,14 @@ object TestNg : TestFramework(displayName = "TestNG") {
         simpleName = "Assert"
     )
 
+    override val arraysAssertionsClass: ClassId = BuiltinClassId(
+        name = TEST_NG_ARRAYS_ASSERTIONS,
+        canonicalName = TEST_NG_ARRAYS_ASSERTIONS,
+        simpleName = "ArrayAsserts"
+    )
+
+    override val assertBooleanArrayEquals by lazy { assertionId("assertEquals", booleanArrayClassId, booleanArrayClassId) }
+
     val throwingRunnableClassId = BuiltinClassId(
         name = "${assertionsClass.name}\$ThrowingRunnable",
         canonicalName = "${assertionsClass.canonicalName}.ThrowingRunnable",
@@ -353,6 +365,7 @@ object Junit4 : TestFramework("JUnit4") {
         canonicalName = JUNIT4_ASSERTIONS,
         simpleName = "Assert"
     )
+    override val arraysAssertionsClass = assertionsClass
 
     val ignoreAnnotationClassId = with("$JUNIT4_PACKAGE.Ignore") {
         BuiltinClassId(
@@ -421,6 +434,8 @@ object Junit5 : TestFramework("JUnit5") {
         simpleName = "Assertions"
     )
 
+    override val arraysAssertionsClass = assertionsClass
+
     val assertThrows = builtinStaticMethodId(
         classId = assertionsClass,
         name = "assertThrows",