chore: Add more tests for direction

Signed-off-by: Tim Hårek Andreassen <tim@harek.no>
This commit is contained in:
Tim Hårek Andreassen 2024-10-02 19:19:36 +02:00
parent dabd75acd9
commit 0faf0f2cf5
No known key found for this signature in database
GPG key ID: E59C7734F0E10EB5
2 changed files with 42 additions and 16 deletions

View file

@ -41,21 +41,23 @@ func (w *wind) DirectionToString() string {
// Retruns unicode block arrow based on wind-direction
func windDirection(d float64) string {
switch {
case d == 0 || d == 360:
case d == 360:
return direction.SOUTH
case d <= 45 && d < 90:
case d >= 0 && d < 45:
return direction.SOUTH
case d >= 45 && d < 90:
return direction.SOUTH_WEST
case d <= 90 && d < 135:
case d >= 90 && d < 135:
return direction.WEST
case d <= 135 && d < 180:
case d >= 135 && d < 180:
return direction.NORTH_WEST
case d <= 180 && d < 225:
case d >= 180 && d < 225:
return direction.NORTH
case d <= 225 && d < 270:
case d >= 225 && d < 270:
return direction.NORTH_EAST
case d <= 270 && d < 315:
case d >= 270 && d < 315:
return direction.EAST
case d <= 315 && d < 360:
case d >= 315:
return direction.SOUTH_EAST
default:
return "↔"

View file

@ -11,26 +11,50 @@ func TestWindDirection(t *testing.T) {
assert := assert.New(t)
south := windDirection(0)
assert.Equal(direction.SOUTH, south, "south")
assert.Equal(direction.SOUTH, south, "south 0")
south = windDirection(20)
assert.Equal(direction.SOUTH, south, "south 20")
southwest := windDirection(45)
assert.Equal(direction.SOUTH_WEST, southwest, "southwest")
assert.Equal(direction.SOUTH_WEST, southwest, "southwest 45")
southwest = windDirection(60)
assert.Equal(direction.SOUTH_WEST, southwest, "southwest 60")
west := windDirection(90)
assert.Equal(direction.WEST, west, "west")
assert.Equal(direction.WEST, west, "west 90")
west = windDirection(100)
assert.Equal(direction.WEST, west, "west 100")
northwest := windDirection(135)
assert.Equal(direction.NORTH_WEST, northwest, "northwest")
assert.Equal(direction.NORTH_WEST, northwest, "northwest 135")
northwest = windDirection(150)
assert.Equal(direction.NORTH_WEST, northwest, "northwest 150")
north := windDirection(180)
assert.Equal(direction.NORTH, north, "north")
assert.Equal(direction.NORTH, north, "north 180")
north = windDirection(200)
assert.Equal(direction.NORTH, north, "north 200")
northeast := windDirection(225)
assert.Equal(direction.NORTH_EAST, northeast, "northeast")
assert.Equal(direction.NORTH_EAST, northeast, "northeast 225")
northeast = windDirection(250)
assert.Equal(direction.NORTH_EAST, northeast, "northeast 250")
east := windDirection(270)
assert.Equal(direction.EAST, east, "east")
assert.Equal(direction.EAST, east, "east 270")
east = windDirection(280)
assert.Equal(direction.EAST, east, "east 280")
southeast := windDirection(315)
assert.Equal(direction.SOUTH_EAST, southeast, "southeast")
assert.Equal(direction.SOUTH_EAST, southeast, "southeast 315")
southeast = windDirection(347.3)
assert.Equal(direction.SOUTH_EAST, southeast, "southeast 347.3")
}