diff options
Diffstat (limited to 'tpl')
-rw-r--r-- | tpl/math/math.go | 86 | ||||
-rw-r--r-- | tpl/math/math_test.go | 156 |
2 files changed, 140 insertions, 102 deletions
diff --git a/tpl/math/math.go b/tpl/math/math.go index 257e803e4..e7460e7ec 100644 --- a/tpl/math/math.go +++ b/tpl/math/math.go @@ -20,7 +20,6 @@ import ( "sync/atomic" _math "github.com/gohugoio/hugo/common/math" - "github.com/spf13/cast" ) @@ -32,9 +31,9 @@ func New() *Namespace { // Namespace provides template functions for the "math" namespace. type Namespace struct{} -// Add adds the two addends n1 and n2. -func (ns *Namespace) Add(n1, n2 any) (any, error) { - return _math.DoArithmetic(n1, n2, '+') +// Add adds the multivalued addends n1 and n2 or more values. +func (ns *Namespace) Add(inputs ...any) (any, error) { + return ns.doArithmetic(inputs, '+') } // Ceil returns the least integer value greater than or equal to n. @@ -48,8 +47,8 @@ func (ns *Namespace) Ceil(n any) (float64, error) { } // Div divides n1 by n2. -func (ns *Namespace) Div(n1, n2 any) (any, error) { - return _math.DoArithmetic(n1, n2, '/') +func (ns *Namespace) Div(inputs ...any) (any, error) { + return ns.doArithmetic(inputs, '/') } // Floor returns the greatest integer value less than or equal to n. @@ -72,28 +71,40 @@ func (ns *Namespace) Log(n any) (float64, error) { return math.Log(af), nil } -// Max returns the greater of the two numbers n1 or n2. -func (ns *Namespace) Max(n1, n2 any) (float64, error) { - af, erra := cast.ToFloat64E(n1) - bf, errb := cast.ToFloat64E(n2) - - if erra != nil || errb != nil { - return 0, errors.New("Max operator can't be used with non-float value") +// Max returns the greater of the multivalued numbers n1 and n2 or more values. +func (ns *Namespace) Max(inputs ...any) (maximum float64, err error) { + var value float64 + for index, input := range inputs { + value, err = cast.ToFloat64E(input) + if err != nil { + err = errors.New("Max operator can't be used with non-float value") + return + } + if index == 0 { + maximum = value + continue + } + maximum = math.Max(value, maximum) } - - return math.Max(af, bf), nil -} - -// Min returns the smaller of two numbers n1 or n2. -func (ns *Namespace) Min(n1, n2 any) (float64, error) { - af, erra := cast.ToFloat64E(n1) - bf, errb := cast.ToFloat64E(n2) - - if erra != nil || errb != nil { - return 0, errors.New("Min operator can't be used with non-float value") + return +} + +// Min returns the smaller of multivalued numbers n1 and n2 or more values. +func (ns *Namespace) Min(inputs ...any) (minimum float64, err error) { + var value float64 + for index, input := range inputs { + value, err = cast.ToFloat64E(input) + if err != nil { + err = errors.New("Max operator can't be used with non-float value") + return + } + if index == 0 { + minimum = value + continue + } + minimum = math.Min(value, minimum) } - - return math.Min(af, bf), nil + return } // Mod returns n1 % n2. @@ -122,9 +133,9 @@ func (ns *Namespace) ModBool(n1, n2 any) (bool, error) { return res == int64(0), nil } -// Mul multiplies the two numbers n1 and n2. -func (ns *Namespace) Mul(n1, n2 any) (any, error) { - return _math.DoArithmetic(n1, n2, '*') +// Mul multiplies the multivalued numbers n1 and n2 or more values. +func (ns *Namespace) Mul(inputs ...any) (any, error) { + return ns.doArithmetic(inputs, '*') } // Pow returns n1 raised to the power of n2. @@ -159,9 +170,20 @@ func (ns *Namespace) Sqrt(n any) (float64, error) { return math.Sqrt(af), nil } -// Sub subtracts n2 from n1. -func (ns *Namespace) Sub(n1, n2 any) (any, error) { - return _math.DoArithmetic(n1, n2, '-') +// Sub subtracts multivalued. +func (ns *Namespace) Sub(inputs ...any) (any, error) { + return ns.doArithmetic(inputs, '-') +} + +func (ns *Namespace) doArithmetic(inputs []any, operation rune) (value any, err error) { + value = inputs[0] + for i := 1; i < len(inputs); i++ { + value, err = _math.DoArithmetic(value, inputs[i], operation) + if err != nil { + return + } + } + return } var counter uint64 diff --git a/tpl/math/math_test.go b/tpl/math/math_test.go index cdec43d5f..076dbbae1 100644 --- a/tpl/math/math_test.go +++ b/tpl/math/math_test.go @@ -26,23 +26,28 @@ func TestBasicNSArithmetic(t *testing.T) { ns := New() - for _, test := range []struct { - fn func(a, b any) (any, error) - a any - b any + type TestCase struct { + fn func(inputs ...any) (any, error) + values []any expect any - }{ - {ns.Add, 4, 2, int64(6)}, - {ns.Add, 1.0, "foo", false}, - {ns.Sub, 4, 2, int64(2)}, - {ns.Sub, 1.0, "foo", false}, - {ns.Mul, 4, 2, int64(8)}, - {ns.Mul, 1.0, "foo", false}, - {ns.Div, 4, 2, int64(2)}, - {ns.Div, 1.0, "foo", false}, + } + + for _, test := range []TestCase{ + {ns.Add, []any{4, 2}, int64(6)}, + {ns.Add, []any{4, 2, 5}, int64(11)}, + {ns.Add, []any{1.0, "foo"}, false}, + {ns.Sub, []any{4, 2}, int64(2)}, + {ns.Sub, []any{4, 2, 5}, int64(-3)}, + {ns.Sub, []any{1.0, "foo"}, false}, + {ns.Mul, []any{4, 2}, int64(8)}, + {ns.Mul, []any{4, 2, 5}, int64(40)}, + {ns.Mul, []any{1.0, "foo"}, false}, + {ns.Div, []any{4, 2}, int64(2)}, + {ns.Div, []any{4, 2, 5}, int64(0)}, + {ns.Div, []any{1.0, "foo"}, false}, } { - result, err := test.fn(test.a, test.b) + result, err := test.fn(test.values...) if b, ok := test.expect.(bool); ok && !b { c.Assert(err, qt.Not(qt.IsNil)) @@ -129,11 +134,11 @@ func TestLog(t *testing.T) { a any expect any }{ - {1, float64(0)}, - {3, float64(1.0986)}, - {0, float64(math.Inf(-1))}, - {1.0, float64(0)}, - {3.1, float64(1.1314)}, + {1, 0.0}, + {3, 1.0986}, + {0, math.Inf(-1)}, + {1.0, 0.0}, + {3.1, 1.1314}, {"abc", false}, } { @@ -170,9 +175,9 @@ func TestSqrt(t *testing.T) { a any expect any }{ - {81, float64(9)}, - {0.25, float64(0.5)}, - {0, float64(0)}, + {81, 9.0}, + {0.25, 0.5}, + {0, 0.0}, {"abc", false}, } { @@ -329,15 +334,15 @@ func TestPow(t *testing.T) { b any expect any }{ - {0, 0, float64(1)}, - {2, 0, float64(1)}, - {2, 3, float64(8)}, - {-2, 3, float64(-8)}, - {2, -3, float64(0.125)}, - {-2, -3, float64(-0.125)}, - {0.2, 3, float64(0.008)}, - {2, 0.3, float64(1.2311)}, - {0.2, 0.3, float64(0.617)}, + {0, 0, 1.0}, + {2, 0, 1.0}, + {2, 3, 8.0}, + {-2, 3, -8.0}, + {2, -3, 0.125}, + {-2, -3, -0.125}, + {0.2, 3, 0.008}, + {2, 0.3, 1.2311}, + {0.2, 0.3, 0.617}, {"aaa", "3", false}, {"2", "aaa", false}, } { @@ -364,28 +369,33 @@ func TestMax(t *testing.T) { ns := New() - for _, test := range []struct { - a any - b any + type TestCase struct { + values []any expect any - }{ - {-1, -1, float64(-1)}, - {-1, 0, float64(0)}, - {-1, 1, float64(1)}, - {0, -1, float64(0)}, - {0, 0, float64(0)}, - {0, 1, float64(1)}, - {1, -1, float64(1)}, - {1, 0, float64(1)}, - {1, 1, float64(1)}, - {1.2, 1.23, float64(1.23)}, - {-1.2, -1.23, float64(-1.2)}, - {0, "a", false}, - {"a", 0, false}, - {"a", "b", false}, - } { + } - result, err := ns.Max(test.a, test.b) + for _, test := range []TestCase{ + // two values + {[]any{-1, -1}, -1.0}, + {[]any{-1, 0}, 0.0}, + {[]any{-1, 1}, 1.0}, + {[]any{0, -1}, 0.0}, + {[]any{0, 0}, 0.0}, + {[]any{0, 1}, 1.0}, + {[]any{1, -1}, 1.0}, + {[]any{1, 0}, 1.0}, + {[]any{1, 1}, 1.0}, + {[]any{1.2, 1.23}, 1.23}, + {[]any{-1.2, -1.23}, -1.2}, + {[]any{0, "a"}, false}, + {[]any{"a", 0}, false}, + {[]any{"a", "b"}, false}, + // multi values + {[]any{-1, -2, -3}, -1.0}, + {[]any{1, 2, 3}, 3.0}, + {[]any{"a", 2, 3}, false}, + } { + result, err := ns.Max(test.values...) if b, ok := test.expect.(bool); ok && !b { c.Assert(err, qt.Not(qt.IsNil)) @@ -403,28 +413,34 @@ func TestMin(t *testing.T) { ns := New() - for _, test := range []struct { - a any - b any + type TestCase struct { + values []any expect any - }{ - {-1, -1, float64(-1)}, - {-1, 0, float64(-1)}, - {-1, 1, float64(-1)}, - {0, -1, float64(-1)}, - {0, 0, float64(0)}, - {0, 1, float64(0)}, - {1, -1, float64(-1)}, - {1, 0, float64(0)}, - {1, 1, float64(1)}, - {1.2, 1.23, float64(1.2)}, - {-1.2, -1.23, float64(-1.23)}, - {0, "a", false}, - {"a", 0, false}, - {"a", "b", false}, + } + + for _, test := range []TestCase{ + // two values + {[]any{-1, -1}, -1.0}, + {[]any{-1, 0}, -1.0}, + {[]any{-1, 1}, -1.0}, + {[]any{0, -1}, -1.0}, + {[]any{0, 0}, 0.0}, + {[]any{0, 1}, 0.0}, + {[]any{1, -1}, -1.0}, + {[]any{1, 0}, 0.0}, + {[]any{1, 1}, 1.0}, + {[]any{1.2, 1.23}, 1.2}, + {[]any{-1.2, -1.23}, -1.23}, + {[]any{0, "a"}, false}, + {[]any{"a", 0}, false}, + {[]any{"a", "b"}, false}, + // multi values + {[]any{-1, -2, -3}, -3.0}, + {[]any{1, 2, 3}, 1.0}, + {[]any{"a", 2, 3}, false}, } { - result, err := ns.Min(test.a, test.b) + result, err := ns.Min(test.values...) if b, ok := test.expect.(bool); ok && !b { c.Assert(err, qt.Not(qt.IsNil)) |