1
0

utils_test.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. package imaging
  2. import (
  3. "runtime"
  4. "testing"
  5. )
  6. func testParallelN(enabled bool, n, procs int) bool {
  7. data := make([]bool, n)
  8. before := runtime.GOMAXPROCS(0)
  9. runtime.GOMAXPROCS(procs)
  10. parallel(n, func(start, end int) {
  11. for i := start; i < end; i++ {
  12. data[i] = true
  13. }
  14. })
  15. for i := 0; i < n; i++ {
  16. if data[i] != true {
  17. return false
  18. }
  19. }
  20. runtime.GOMAXPROCS(before)
  21. return true
  22. }
  23. func TestParallel(t *testing.T) {
  24. for _, e := range []bool{true, false} {
  25. for _, n := range []int{1, 10, 100, 1000} {
  26. for _, p := range []int{1, 2, 4, 8, 16, 100} {
  27. if testParallelN(e, n, p) != true {
  28. t.Errorf("test [parallel %v %d %d] failed", e, n, p)
  29. }
  30. }
  31. }
  32. }
  33. }
  34. func TestClamp(t *testing.T) {
  35. td := []struct {
  36. f float64
  37. u uint8
  38. }{
  39. {0, 0},
  40. {255, 255},
  41. {128, 128},
  42. {0.49, 0},
  43. {0.50, 1},
  44. {254.9, 255},
  45. {254.0, 254},
  46. {256, 255},
  47. {2500, 255},
  48. {-10, 0},
  49. {127.6, 128},
  50. }
  51. for _, d := range td {
  52. if clamp(d.f) != d.u {
  53. t.Errorf("test [clamp %v %v] failed: %v", d.f, d.u, clamp(d.f))
  54. }
  55. }
  56. }
  57. func TestClampint32(t *testing.T) {
  58. td := []struct {
  59. i int32
  60. u uint8
  61. }{
  62. {0, 0},
  63. {255, 255},
  64. {128, 128},
  65. {256, 255},
  66. {2500, 255},
  67. {-10, 0},
  68. }
  69. for _, d := range td {
  70. if clampint32(d.i) != d.u {
  71. t.Errorf("test [clampint32 %v %v] failed: %v", d.i, d.u, clampint32(d.i))
  72. }
  73. }
  74. }