最近在看《敏捷軟體開發》，看到第31的有關於編寫測試類測試一個素數生成器的例子，特此動手實踐了下，發現了書中的一個小錯誤，同時在網上看到一個更好的測試方法。

版本一

//這是書上版本一
public class GeneratePrimes {
    public static int[] generatePrimes(int maxValue){
        if(maxValue >=2){
            int s = maxValue+1;
            boolean[] f = new boolean[s];
            int i;

            for
 
(i=0;i<s;i++){
                f[i]=true;
            }

            f[0]=f[1]=false;

            int j;

            for(i=2;i<Math.sqrt(s)+1;i++){
                for(j=2*i;j<s;j+=i)
                    f[j] = false;
            }

            int count=0;

            for(i=0;i<s;i++){
                if
 
(f[i])
                    count++;
            }
            int[] primes = new int[count];

            for(i=0,j=0;i<s;i++){
                if(f[i])
                    primes[j++]=i;
            }
            return primes;
        }
        else {
            return new int[0];
        }
    }
}
 

版本二

//這是書上版本二

public class PrimeGenerator {
    private static int s;//這裡書上是a，很明顯錯了
    private static boolean[] f;
    private static int[] primes;

    public static int[] generatePrimes(int maxValue){
        if(maxValue<2){
            return new int[0];
        }else{
            initializeSieve(maxValue);
            sieve();
            loadPrimes();
            return primes;
        }
    }


    private static void loadPrimes() {

        int i;
        int j;

        int count = 0;
        for(i=0;i<s;i++){
            if(f[i]){
                count++;
            }
        }

        primes = new int[count];

        for(i=0,j=0;i<s;i++){
            if(f[i]){
                primes[j++]=i;
            }
        }

    }

    private static void sieve() {

        int i;
        int j;
        for(i=2;i<Math.sqrt(s)+1;i++){
            if(f[i]){
                for(j=2*i;j<s;j+=i){
                    f[j]=false;
                }
            }
        }

    }

    private static void initializeSieve(int maxValue) {

        s=maxValue+1;
        f=new boolean[s];
        int i;

        for(i=0;i<s;i++){
            f[i]=true;
        }

        f[0]=f[1]=false;

    }
}

書上的測試類

測試類，書上的測試PrimeGenerator的，順著寫就好

import junit.framework.TestCase;
import junit.textui.TestRunner;

public class TestGeneratePrimes2 extends TestCase {

    public static void main(String[] args) {

        TestRunner.main(new String[]{"TestGeneratePrimes"});
    }

    public TestGeneratePrimes2(String name){
        super(name);
    }

    public void testPrimes(){
        int[] nullArray = PrimeGenerator.generatePrimes(0);
        assertEquals(nullArray.length, 0);

        int[] minArray = PrimeGenerator.generatePrimes(2);
        assertEquals(minArray.length, 1);
        assertEquals(minArray[0], 2);

        int[] threeArray = PrimeGenerator.generatePrimes(3);
        assertEquals(threeArray.length, 2);
        assertEquals(threeArray[0], 2);
        assertEquals(threeArray[1], 3);

        int[] centArray = PrimeGenerator.generatePrimes(100);
        assertEquals(centArray.length, 25);
        assertEquals(centArray[24], 97);


    }

}

我自己寫的測試類

測試類，我自己寫來測試GeneratePrimes的

import static org.junit.Assert.*;

import org.junit.Test;

public class TestGeneratePrimes {

    @Test
    public void test() {
        int[] nullArray = GeneratePrimes.generatePrimes(0);
        assertEquals(nullArray.length, 0);

        int[] minArray = GeneratePrimes.generatePrimes(2);
        assertEquals(minArray.length, 1);
        assertEquals(minArray[0], 2);

        int[] threeArray = GeneratePrimes.generatePrimes(3);
        assertEquals(threeArray.length, 2);
        assertEquals(threeArray[0], 2);
        assertEquals(threeArray[1], 3);

        int[] centArray = GeneratePrimes.generatePrimes(100);
        assertEquals(centArray.length, 25);
        assertEquals(centArray[24], 97);
    }

}

可以發現程式碼明顯少了很多。

或者這樣

import static org.junit.Assert.*;

import org.junit.Test;

public class TestGeneratePrimes {

    @Test
    public void testNull() {
        int[] nullArray = GeneratePrimes.generatePrimes(0);
        assertEquals(nullArray.length, 0);
    }

    @Test
    public void testMin(){
        int[] minArray = GeneratePrimes.generatePrimes(2);
        assertEquals(minArray.length, 1);
        assertEquals(minArray[0], 2);
    }

    @Test
    public void testThree(){
        int[] threeArray = GeneratePrimes.generatePrimes(3);
        assertEquals(threeArray.length, 2);
        assertEquals(threeArray[0], 2);
        assertEquals(threeArray[1], 3);
    }

    @Test
    public void testCent(){
        int[] centArray = GeneratePrimes.generatePrimes(100);
        assertEquals(centArray.length, 25);
        assertEquals(centArray[24], 97);
    }
}

測試結果看著也很清爽。

就是醬紫。