tweaks

example/example.go

@@ -2,6 +2,8 @@ package example
 
 import "context"
 
+//go:generate moq PersonStore -out mockpersonstore_test.go
+
 type Person struct {
 	ID      string
 	Name    string

package/moq/moq.go

@@ -78,9 +78,7 @@ func (m *Mocker) Mock(w io.Writer, name ...string) error {
 				method.Returns = extractArgs(sig.Results(), "out%d")
 			}
 			objs = append(objs, obj)
-
 		}
-
 	}
 	err := m.tmpl.Execute(w, struct{ Objs []*obj }{Objs: objs})
 	if err != nil {

package/moq/moq_test.go

@@ -1,16 +1,14 @@
-package moq_test
+package moq
 
 import (
 	"bytes"
 	"log"
 	"testing"
-
-	"github.com/matryer/moq/package/moq"
 )
 
 func TestMoq(t *testing.T) {
 
-	m, err := moq.New("../../example")
+	m, err := New("../../example")
 	if err != nil {
 		t.Errorf("moq.New: %s", err)
 	}