package entity

import (
	"testing"

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

func TestLocation_Label(t *testing.T) {
	l := NewLocation(1, 1)
	l.LocCategory = "restaurant"
	result := l.Category()

	assert.Equal(t, "restaurant", result)
}