package main

import (
	student "student"

	"github.com/01-edu/public/go/tests/lib"
	"github.com/01-edu/public/go/tests/lib/is"
)

func isPositive(i int) bool {
	return i > 0
}

func isNegative(i int) bool {
	return i < 0
}

func _map(f func(int) bool, a []int) []bool {
	result := make([]bool, len(a))

	for i, el := range a {
		result[i] = f(el)
	}

	return result
}

func main() {
	functions := []func(int) bool{isPositive, isNegative, is.Prime}

	type node struct {
		f func(int) bool
		a []int
	}

	table := []node{}

	for i := 0; i < 15; i++ {
		function := functions[lib.RandIntBetween(0, len(functions)-1)]
		val := node{
			f: function,
			a: lib.MultRandIntBetween(-1000000, 1000000),
		}
		table = append(table, val)
	}

	table = append(table, node{
		f: is.Prime,
		a: []int{1, 2, 3, 4, 5, 6},
	})

	for _, arg := range table {
		lib.Challenge("Map", student.Map, _map, arg.f, arg.a)
	}
}