public/go/tests/func/trimatoi_test/main.go

package main

import (
	student "student"

	"lib"
)

func trimAtoi(s string) int {
	runes := []rune(s)
	for i := range s {
		runes[i] = rune(s[i])
	}
	var numbers []rune
	for _, r := range runes {
		if (r >= '0' && r <= '9') || (len(numbers) == 0 && (r) == '-' || r == '+') {
			numbers = append(numbers, r)
		}
	}
	if len(numbers) == 0 || (len(numbers) == 1 && (numbers[0] == '-' || numbers[0] == '+')) {
		return 0
	}

	res, i, sign := 0, 0, 1

	if numbers[0] == '-' {
		sign = -1
		i++
	} else if numbers[0] == '+' {
		i++
	}
	for ; i < len(numbers); i++ {
		res = res*10 + int(numbers[i]) - '0'
	}

	return sign * res
}

func stringsToTrimAtoi(a []string) []string {
	alpha := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789012345678901234567890123456789"

	for index := 0; index < 4; index++ {
		s := ""
		s += lib.RandStr(lib.RandIntBetween(0, 2), alpha)
		x := lib.RandIntBetween(0, 14)
		if x <= 4 {
			s += "-"
		}
		s += lib.RandStr(lib.RandIntBetween(0, 10), alpha)
		a = append(a, s)
	}
	return a
}

func main() {
	a := []string{
		"",
		"12345",
		"str123ing45",
		"012 345",
		"Hello World!",
		"sd+x1fa2W3s4",
		"sd-x1fa2W3s4",
		"sdx1-fa2W3s4",
		lib.RandAlnum(),
	}
	a = stringsToTrimAtoi(a)
	for _, elem := range a {
		lib.Challenge("TrimAtoi", student.TrimAtoi, trimAtoi, elem)
	}
}

// TODO: refactor, including the subject