Add test for units

pkg/db/units_test.go

@@ -0,0 +1,47 @@
+package db_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	. "recipe_book/pkg/db"
+)
+
+func TestUnitsNamesAndAbbrevs(t *testing.T) {
+	assert := assert.New(t)
+	assert.Equal("count", COUNT.Name())
+	assert.Equal("cups", CUPS.Name())
+	assert.Equal("pounds", LBS.Name())
+
+	assert.Equal("tsp", TSP.Abbreviation())
+	assert.Equal("fl-oz", FLOZ.Abbreviation())
+}
+
+func TestUnitsOfFoods(t *testing.T) {
+	assert := assert.New(t)
+	db := get_test_db()
+
+	avocado := get_food(db, 20)
+	onion := get_food(db, 28)
+	eggs := get_food(db, 8)
+	oats := get_food(db, 2)
+
+	ingr := COUNT.Of(avocado, 3)
+	assert.Equal(ingr.FoodID, FoodID(20))
+	assert.Equal(ingr.Quantity, float32(3.0))
+	assert.Equal(ingr.Units, COUNT)
+
+	ingr = GRAMS.Of(avocado, 50)
+	assert.Equal(ingr.Quantity, float32(0.5))
+	assert.Equal(ingr.Units, GRAMS)
+
+	assert.Equal(GRAMS.Of(eggs, 106).Quantity, float32(2))
+	assert.InDelta(LBS.Of(onion, 2).Quantity, float32(4.127), 0.001)
+	assert.InDelta(OZ.Of(onion, 2).Quantity, float32(0.254), 0.001)
+	assert.InDelta(ML.Of(onion, 200).Quantity, float32(0.909), 0.001)
+	assert.InDelta(ML.Of(oats, 200).Quantity, float32(0.88), 0.001)
+	assert.InDelta(CUPS.Of(oats, 2).Quantity, float32(2.2), 0.001)
+	assert.InDelta(TSP.Of(avocado, 1.5).Quantity, float32(0.075), 0.001)
+	assert.InDelta(TBSP.Of(avocado, 1.5).Quantity, float32(0.225), 0.001)
+}