如何在同一文件中为Kotlin代码添加单元测试并运行？

嘿，我来帮你搞定这个问题！在Kotlin里，你完全可以像Python那样在同一个文件里写测试代码，然后直接触发运行。下面给你两种常用的实现方式：

方式一：使用JUnit 5做正规单元测试（推荐） #

这种方式和Python的unittest框架逻辑类似，能生成详细的测试报告，适合项目级的规范测试。你只需要把业务类和测试类放在同一个文件，再通过main函数触发测试执行。

完整代码示例 #

import org.junit.jupiter.api.Test
import org.junit.jupiter.api.Assertions.assertEquals
import org.junit.platform.launcher.core.LauncherFactory
import org.junit.platform.engine.discovery.DiscoverySelectors.selectClass

// 你的业务实现类
class Solution {
    fun strStr(haystack: String, needle: String): Int {
        return haystack.indexOf(needle)
    }
}

// 测试类
class SolutionTest {
    private val solution = Solution()

    @Test
    fun `test basic match case`() {
        assertEquals(2, solution.strStr("hello", "ll"))
    }

    @Test
    fun `test empty needle`() {
        assertEquals(0, solution.strStr("aaaaa", ""))
    }

    @Test
    fun `test needle not found`() {
        assertEquals(-1, solution.strStr("aaaaa", "bba"))
    }
}

// 触发测试的入口，对应Python的if __name__ == '__main__'
fun main() {
    val launcher = LauncherFactory.create()
    launcher.execute(selectClass(SolutionTest::class.java))
}

关键说明 #

1. 依赖准备：如果用Gradle/Maven构建项目，需要添加JUnit 5的依赖。比如Gradle的build.gradle.kts中添加：

dependencies {
    testImplementation("org.junit.jupiter:junit-jupiter-api:5.9.2")
    testRuntimeOnly("org.junit.jupiter:junit-jupiter-engine:5.9.2")
}

用IntelliJ的话，它会自动检测缺失的依赖并提示你一键添加。

2. 测试方法编写：用@Test注解标记测试方法，用反引号包裹方法名可以让测试名称更直观（比如test basic match case）。assertEquals用来验证结果是否符合预期。

3. 运行测试：直接运行这个文件的main函数，JUnit会自动执行所有测试方法，并在控制台输出详细的测试结果（通过/失败数量、失败原因等）。

方式二：简单断言测试（快速验证） #

如果只是快速验证代码逻辑，不想引入额外依赖，可以直接在main函数里写断言：

class Solution {
    fun strStr(haystack: String, needle: String): Int {
        return haystack.indexOf(needle)
    }
}

fun main() {
    val solution = Solution()
    
    // 逐个验证测试用例
    assert(solution.strStr("hello", "ll") == 2) { "Basic match case failed!" }
    assert(solution.strStr("aaaaa", "") == 0) { "Empty needle case failed!" }
    assert(solution.strStr("aaaaa", "bba") == -1) { "Needle not found case failed!" }
    
    println("All tests passed successfully!")
}

这种方式更轻量化，运行后如果所有断言都通过，会打印成功信息；如果有失败，会抛出AssertionError并显示你自定义的错误提示。

内容的提问来源于stack exchange，提问作者Muhammad Lukman Low