chore: Add more tests for direction
Signed-off-by: Tim Hårek Andreassen <tim@harek.no>
This commit is contained in:
parent
dabd75acd9
commit
0faf0f2cf5
2 changed files with 42 additions and 16 deletions
18
yr/yr.go
18
yr/yr.go
|
@ -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 "↔"
|
||||
|
|
|
@ -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")
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue