fiddlek/go/linenumbers/linenumbers_test.go

package linenumbers

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestLineNumbers(t *testing.T) {
	code := `a
b
c`
	want := `#line 1
a
b
c`
	assert.Equal(t, want, LineNumbers(code))

	code = ``
	want = `#line 1
`
	assert.Equal(t, want, LineNumbers(code))
}