请给出源码:junit判断是否是素数,参考数据集是 {true,1},{true,7},给可以给出其他数据测试。
时间: 2024-05-01 13:18:21 浏览: 45
当然,这是一个很简单的问题,您可以参考以下示例代码来判断一个数是否为素数:
```
import org.junit.Test;
import static org.junit.Assert.assertEquals;
public class PrimeNumberTest {
@Test
public void testIsPrime() {
assertEquals(true, isPrime(1));
assertEquals(true, isPrime(7));
assertEquals(false, isPrime(4));
assertEquals(false, isPrime(20));
}
public boolean isPrime(int n) {
if (n <= 1) {
return false;
}
for (int i = 2; i <= Math.sqrt(n); i++) {
if (n % i == 0) {
return false;
}
}
return true;
}
}
```
上面的代码可以判断输入的数是否为素数,其中 `isPrime()` 方法接收一个整数 `n`,通过遍历所有小于 `n` 的数来判断是否存在除了1和自身之外其它的因子。
我们在测试中可以用 `assertEquals()` 方法来判断期望的值和实际值是否相等,可以很方便地将多组测试数据加入测试中,如上述代码中的 `assertEquals(true, isPrime(1))` 和 `assertEquals(true, isPrime(7))`。
希望我的回答能够帮到您!
阅读全文