diff --git a/draw_test.go b/draw_test.go index 0c7888a..75afc83 100644 --- a/draw_test.go +++ b/draw_test.go @@ -774,7 +774,11 @@ s.trimSpace = tc.trim s.refill = tc.refill tmpBuf.Reset() - _, err := tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + r, err := s.draw(newStatistics(tw, s)) + if err != nil { + t.FailNow() + } + _, err = tmpBuf.ReadFrom(r) if err != nil { t.FailNow() } @@ -1224,7 +1228,11 @@ s.trimSpace = tc.trim s.refill = tc.refill tmpBuf.Reset() - _, err := tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + r, err := s.draw(newStatistics(tw, s)) + if err != nil { + t.FailNow() + } + _, err = tmpBuf.ReadFrom(r) if err != nil { t.FailNow() } @@ -1396,7 +1404,11 @@ s.trimSpace = tc.trim s.refill = tc.refill tmpBuf.Reset() - _, err := tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + r, err := s.draw(newStatistics(tw, s)) + if err != nil { + t.FailNow() + } + _, err = tmpBuf.ReadFrom(r) if err != nil { t.FailNow() }