자바 유닛 테스트 코드를 적용한 알고리즘 풀이 연습 저장소
그럼 이렇게 유닛 테스트 파일이 생김
SoultuonTest.java
package wee1.sol1;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
public class SolutionTest {
@Test
public void testSolution(){
Solution sol = new Solution();
Assertions.assertEquals(1,sol.solution(new int[] {1,2,3,4}));
Assertions.assertEquals(4,sol.solution(new int[] {1,2,7,6,4}));
}
}
그 후 Assertions.assertEquals(<정답 리턴 결과>,<테스트할 메소드>);
으로 테스트 케이스 세팅
입력 | 출력 |
---|---|
[1,2,3,4] |
1 |
[1,2,7,6,4] |
1 |
[...] |
... |
다시 Solutin.java로 돌아와서 테스트할 메소드 생성
package wee1.sol1;
public class Solution {
public int solution(int[] nums) {
int answer = 0;
for(int i = 0 ;i < nums.length-2; i++){
for(int j = i + 1; j<nums.length-1; j++){
for(int k = j + 1; k <nums.length; k++){
int sum = nums[i] + nums[j] + nums[k];
if(isPrime(sum)){
answer+=1;
}
}
}
}
return answer;
}
private boolean isPrime(int sum) {
for(int i = 2; i<sum; i++){
if(sum % i == 0) return false;
}
return true;
}
}
실행해서 테스트 통과 여부 확인