bencode: Tidy up

bencode/encode.go

@@ -1,11 +1,13 @@
 package bencode
 
-import "bufio"
-import "reflect"
-import "runtime"
-import "strconv"
-import "sync"
-import "sort"
+import (
+	"bufio"
+	"reflect"
+	"runtime"
+	"sort"
+	"strconv"
+	"sync"
+)
 
 func is_empty_value(v reflect.Value) bool {
 	switch v.Kind() {
@@ -138,7 +140,6 @@ func (e *encoder) reflect_value(v reflect.Value) {
 			if ef.omit_empty && is_empty_value(field_value) {
 				continue
 			}
-
 			e.reflect_string(ef.tag)
 			e.reflect_value(field_value)
 		}

bencode/encode_test.go

@@ -1,6 +1,10 @@
 package bencode
 
-import "testing"
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
 import "bytes"
 import "fmt"
 
@@ -57,12 +61,7 @@ var random_encode_tests = []random_encode_test{
 func TestRandomEncode(t *testing.T) {
 	for _, test := range random_encode_tests {
 		data, err := Marshal(test.value)
-		if err != nil {
-			t.Fatal(err)
-		}
-		if !bytes.Equal(data, []byte(test.expected)) {
-			t.Errorf("got: %s, expected: %s\n",
-				string(data), string(test.expected))
-		}
+		assert.NoError(t, err)
+		assert.EqualValues(t, test.expected, string(data))
 	}
 }