Added arg parsing and test case, fixed an ungeneralized 18

1 files changed, 24 insertions, 0 deletions

diff --git a/capset_test.go b/capset_test.go
index cd2715e..f2693ca 100644
--- a/capset_test.go
+++ b/capset_test.go
@@ -4,6 +4,9 @@ import (
 	"fmt"
 	"reflect"
 	"testing"
+	"bufio"
+	"log"
+	"os"
 )
 
 var deck = NewDeck(4)
@@ -76,6 +79,27 @@ func TestNoAllocs(t *testing.T) {
 	}
 }
 
+func TestDim3(t *testing.T) {
+	d := NewDeck(3)
+	cs := NewCapset(d, -1)
+
+	cmpfile, err := os.Open("./capset3.out")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer cmpfile.Close()
+	scanner := bufio.NewScanner(cmpfile)
+
+	for cs.FindNextCapset() {
+		scanner.Scan()
+		newline := fmt.Sprint(cs.GetTableau())
+		if newline != scanner.Text() {
+			t.Errorf("Program returned %v\n capset3.out has %v",
+				newline, scanner.Text())
+		}
+	}
+}
+
 func BenchmarkIncrement(b *testing.B) {
 	cs := NewCapset(deck, -1)
 	for i := 0; i < b.N; i++ {