32 lines
549 B
Go
32 lines
549 B
Go
package trapping_rain_water
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTrap_Example1(t *testing.T) {
|
|
height := []int{0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1}
|
|
expected := 6
|
|
result := Trap(height)
|
|
|
|
assert.Equal(t, expected, result)
|
|
}
|
|
|
|
func TestTrap_Example2(t *testing.T) {
|
|
height := []int{4, 2, 0, 3, 2, 5}
|
|
expected := 9
|
|
result := Trap(height)
|
|
|
|
assert.Equal(t, expected, result)
|
|
}
|
|
|
|
func TestTrap_Example3(t *testing.T) {
|
|
height := []int{0}
|
|
expected := 0
|
|
result := Trap(height)
|
|
|
|
assert.Equal(t, expected, result)
|
|
}
|