Consistently use github.com/stretchr/testify/require

Instead of writing:

```go
if !assert.Equal(t, want, got) {
        return
}
```

We should probably just write:

```go
require.Equal(t, want, got)
```

This will stop the test if `want` is not equal to `got`, which is probably what we want in most cases.
