Small improvements to PR

This commit is contained in:
Ivan Daniluk 2015-09-25 12:39:31 -04:00
parent b33870f09c
commit 34398877f7
2 changed files with 12 additions and 10 deletions

View File

@ -33,13 +33,13 @@ func (s *Stack) Push(val VarValue) {
s.Values = s.Values[1:]
}
if s.Max == nil {
s.Max = val
return
}
switch val.(type) {
case int64:
if s.Max == nil {
s.Max = val
return
}
switch s.Max.(type) {
case int64:
if val.(int64) > s.Max.(int64) {
@ -51,11 +51,6 @@ func (s *Stack) Push(val VarValue) {
}
}
case float64:
if s.Max == nil {
s.Max = val
return
}
switch s.Max.(type) {
case int64:
if val.(float64) > float64(s.Max.(int64)) {

View File

@ -7,6 +7,13 @@ func TestPushWithFloatAndIntValue(t *testing.T) {
s.Push(VarValue(int64(0.0))) // from service.go:guessValue
s.Push(VarValue(5.0))
s.Push(VarValue(float64(15.0)))
if _, ok := s.Max.(float64); !ok {
t.Fatalf("Expected Max to be float64, but it's not")
}
s.Push(VarValue(int64(25.0)))
if _, ok := s.Max.(int64); !ok {
t.Fatalf("Expected Max to be int64, but it's not")
}
}
func TestStack(t *testing.T) {