From 30449c75ec96dfb5f84b3b397496349c29ab6631 Mon Sep 17 00:00:00 2001 From: Jim Myhrberg Date: Sun, 3 Jul 2016 00:21:11 +0100 Subject: [PATCH] Clean up tests a bit --- base58_test.go | 17 +++++------------ 1 file changed, 5 insertions(+), 12 deletions(-) diff --git a/base58_test.go b/base58_test.go index 999ed36..0f4b8ae 100644 --- a/base58_test.go +++ b/base58_test.go @@ -3,7 +3,6 @@ package base58 import ( "testing" - "github.com/stretchr/testify/assert" "github.com/stretchr/testify/suite" ) @@ -182,34 +181,28 @@ type Base58Suite struct { func (s *Base58Suite) TestAlphabet() { expected := "123456789abcdefghijkmnopqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ" - assert.Equal(s.T(), expected, Alphabet) + s.Equal(expected, Alphabet) } func (s *Base58Suite) TestEncode() { - assert := assert.New(s.T()) - for str, num := range examples { result := Encode(num) - assert.Equal(str, result) + s.Equal(str, result) } } func (s *Base58Suite) TestDecode() { - assert := assert.New(s.T()) - for str, num := range examples { result, _ := Decode(str) - assert.Equal(num, result) + s.Equal(num, result) } } func (s *Base58Suite) TestDecodeError() { - assert := assert.New(s.T()) - result, err := Decode("invalid@base58.string") - assert.Equal(-1, result) - assert.Equal("invalid base58", err.Error()) + s.Equal(-1, result) + s.Equal("invalid base58", err.Error()) } func TestBase58Suite(t *testing.T) {