iter_test.go

package pretty_test

import (
	"iter"
	"maps"
	"slices"

	. "github.com/pierrre/pretty"
)

func init() {
	addTestCasesPrefix("IterSeq", []*testCase{
		{
			name:  "Default",
			value: slices.Values([]string{"a", "b", "c"}),
		},
		{
			name:  "Nil",
			value: iter.Seq[int](nil),
		},
		{
			name:  "Empty",
			value: slices.Values([]string(nil)),
		},
		{
			name:  "Truncated",
			value: slices.Values([]string{"a", "b", "c"}),
			configure: func(vw *CommonValueWriter) {
				vw.Iter.MaxLen = 2
			},
		},
		{
			name: "Large",
			value: func() iter.Seq[int] {
				i := 0
				return func(yield func(int) bool) {
					for {
						if i >= 100 {
							return
						}
						if !yield(i) {
							return
						}
						i++
					}
				}
			}(),
		},
		{
			name:  "Disabled",
			value: slices.Values([]string{"a", "b", "c"}),
			configure: func(vw *CommonValueWriter) {
				vw.Iter = nil
			},
		},
	})
	addTestCasesPrefix("IterSeq2", []*testCase{
		{
			name:  "Default",
			value: slices.All([]string{"a", "b", "c"}),
		},
		{
			name:  "Nil",
			value: iter.Seq2[string, int](nil),
		},
		{
			name:  "Empty",
			value: slices.All([]string(nil)),
		},
		{
			name:  "KeysStringShowInfos",
			value: maps.All(map[string]int{"a": 1}),
			configure: func(vw *CommonValueWriter) {
				vw.Iter.ShowKeysInfos = true
			},
		},
		{
			name: "Large",
			value: func() iter.Seq2[int, int] {
				i := 0
				return func(yield func(int, int) bool) {
					for {
						if i >= 100 {
							return
						}
						if !yield(i, i) {
							return
						}
						i++
					}
				}
			}(),
		},
		{
			name:  "Disabled",
			value: slices.All([]string{"a", "b", "c"}),
			configure: func(vw *CommonValueWriter) {
				vw.Iter = nil
			},
		},
	})
}