From 6abca706a944a62231608c4a8d8fbdff81f4ca3c Mon Sep 17 00:00:00 2001 From: Nick White Date: Mon, 13 May 2019 19:46:27 +0100 Subject: Add -slow flag to test to skip slow tests by default --- preproc/helpers_test.go | 56 ------------------------------------------------ preproc/sauvola_test.go | 6 +++++- preproc/test_helpers.go | 57 +++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 62 insertions(+), 57 deletions(-) delete mode 100644 preproc/helpers_test.go create mode 100644 preproc/test_helpers.go diff --git a/preproc/helpers_test.go b/preproc/helpers_test.go deleted file mode 100644 index 326b59d..0000000 --- a/preproc/helpers_test.go +++ /dev/null @@ -1,56 +0,0 @@ -package preproc - -// TODO: add different pages as test cases -// TODO: test non integral img version - -import ( - "flag" - "image" - "image/draw" - "image/png" - "os" -) - -var update = flag.Bool("update", false, "update golden files") - -func decode(s string) (*image.Gray, error) { - f, err := os.Open(s) - defer f.Close() - if err != nil { - return nil, err - } - img, err := png.Decode(f) - if err != nil { - return nil, err - } - b := img.Bounds() - gray := image.NewGray(image.Rect(0, 0, b.Dx(), b.Dy())) - draw.Draw(gray, b, img, b.Min, draw.Src) - return gray, nil -} - -func imgsequal(img1 *image.Gray, img2 *image.Gray) bool { - b := img1.Bounds() - if !b.Eq(img2.Bounds()) { - return false - } - for y := b.Min.Y; y < b.Max.Y; y++ { - for x := b.Min.X; x < b.Max.X; x++ { - r0, g0, b0, a0 := img1.At(x, y).RGBA() - r1, g1, b1, a1 := img2.At(x, y).RGBA() - if r0 != r1 { - return false - } - if g0 != g1 { - return false - } - if b0 != b1 { - return false - } - if a0 != a1 { - return false - } - } - } - return true -} diff --git a/preproc/sauvola_test.go b/preproc/sauvola_test.go index 1397a4f..5582941 100644 --- a/preproc/sauvola_test.go +++ b/preproc/sauvola_test.go @@ -35,7 +35,11 @@ func TestBinarization(t *testing.T) { case "integralsauvola": actual = IntegralSauvola(orig, c.ksize, c.wsize) case "sauvola": - actual = Sauvola(orig, c.ksize, c.wsize) + if *slow { + actual = Sauvola(orig, c.ksize, c.wsize) + } else { + t.Skip("Skipping slow test; use -slow to run it.\n") + } default: t.Fatalf("No method %s\n", c.name) } diff --git a/preproc/test_helpers.go b/preproc/test_helpers.go new file mode 100644 index 0000000..7579e88 --- /dev/null +++ b/preproc/test_helpers.go @@ -0,0 +1,57 @@ +package preproc + +// TODO: add different pages as test cases +// TODO: test non integral img version + +import ( + "flag" + "image" + "image/draw" + "image/png" + "os" +) + +var slow = flag.Bool("slow", false, "include slow tests") +var update = flag.Bool("update", false, "update golden files") + +func decode(s string) (*image.Gray, error) { + f, err := os.Open(s) + defer f.Close() + if err != nil { + return nil, err + } + img, err := png.Decode(f) + if err != nil { + return nil, err + } + b := img.Bounds() + gray := image.NewGray(image.Rect(0, 0, b.Dx(), b.Dy())) + draw.Draw(gray, b, img, b.Min, draw.Src) + return gray, nil +} + +func imgsequal(img1 *image.Gray, img2 *image.Gray) bool { + b := img1.Bounds() + if !b.Eq(img2.Bounds()) { + return false + } + for y := b.Min.Y; y < b.Max.Y; y++ { + for x := b.Min.X; x < b.Max.X; x++ { + r0, g0, b0, a0 := img1.At(x, y).RGBA() + r1, g1, b1, a1 := img2.At(x, y).RGBA() + if r0 != r1 { + return false + } + if g0 != g1 { + return false + } + if b0 != b1 { + return false + } + if a0 != a1 { + return false + } + } + } + return true +} -- cgit v1.2.1-24-ge1ad