Code improvements

Author: EgorkaKulikov

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgMethodConstructor.kt

@@ -3,8 +3,13 @@ package org.utbot.framework.codegen.model.constructor.tree
 import org.utbot.common.PathUtil
 import org.utbot.common.packageName
 import org.utbot.engine.isStatic
-import org.utbot.framework.codegen.*
+import org.utbot.framework.codegen.ForceStaticMocking
+import org.utbot.framework.codegen.JUNIT5_PARAMETERIZED_PACKAGE
+import org.utbot.framework.codegen.Junit4
+import org.utbot.framework.codegen.Junit5
+import org.utbot.framework.codegen.ParametrizedTestSource
 import org.utbot.framework.codegen.RuntimeExceptionTestsBehaviour.PASS
+import org.utbot.framework.codegen.TestNg
 import org.utbot.framework.codegen.model.constructor.builtin.closeMethodIdOrNull
 import org.utbot.framework.codegen.model.constructor.builtin.forName
 import org.utbot.framework.codegen.model.constructor.builtin.getClass
@@ -1043,12 +1048,13 @@ internal class CgMethodConstructor(val context: CgContext) : CgContextOwner by c
             ParametrizedTestSource.DO_NOT_PARAMETRIZE ->
                 currentBlock = currentBlock.addAll(generateDeepEqualsAssertion(expected, actual))
             ParametrizedTestSource.PARAMETRIZE -> {
-                val trueStmts = listOf<CgStatement>(
-                    testFrameworkManager.assertions[testFramework.assertNull](actual).toStatement()
-                )
-                val falseStmts = generateDeepEqualsAssertion(expected, actual)
+                val assertNullStmt = listOf(testFrameworkManager.assertions[testFramework.assertNull](actual).toStatement())
                 currentBlock = currentBlock.add(
-                    CgIfStatement(CgEqualTo(expected, nullLiteral()), trueStmts, falseStmts)
+                    CgIfStatement(
+                        CgEqualTo(expected, nullLiteral()),
+                        assertNullStmt,
+                        generateDeepEqualsAssertion(expected, actual)
+                    )
                 )
             }
         }