From 52c14808c69f2c657226b81d6bb924bd115e3a9e Mon Sep 17 00:00:00 2001
From: shixzie <shixzie@gmail.com>
Date: Thu, 3 Aug 2017 12:27:45 -0430
Subject: [PATCH] simple benchmark

---
 benchmark_test.go | 111 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 111 insertions(+)
 create mode 100644 benchmark_test.go

diff --git a/benchmark_test.go b/benchmark_test.go
new file mode 100644
index 0000000..baa9f83
--- /dev/null
+++ b/benchmark_test.go
@@ -0,0 +1,111 @@
+package nlp
+
+import (
+	"testing"
+	"time"
+)
+
+func BenchmarkNL_P(b *testing.B) {
+	type T struct {
+		String string
+		Int    int
+		Uint   uint
+		Float  float32
+		Time   time.Time
+		Dur    time.Duration
+	}
+
+	tSamples := []string{
+		"string {String}",
+		"int {Int}",
+		"uint {Uint}",
+		"float {Float}",
+		"time {Time}",
+		"dur {Dur}",
+		"string {String} int {Int}",
+		"string {String} time {Time}",
+	}
+
+	nl := New()
+
+	nl.RegisterModel(T{}, tSamples)
+
+	err := nl.RegisterModel(T{}, tSamples)
+	if err != nil {
+		b.Error(err)
+	}
+
+	err = nl.Learn()
+	if err != nil {
+		b.Error(err)
+	}
+
+	tim, err := time.ParseInLocation("01-02-2006_3:04pm", "05-18-1999_6:42pm", time.Local)
+	if err != nil {
+		b.Error(err)
+	}
+
+	dur, err := time.ParseDuration("4h2m")
+	if err != nil {
+		b.Error(err)
+	}
+
+	cases := []struct {
+		name       string
+		expression string
+		want       interface{}
+	}{
+		{
+			"string",
+			"string Hello World",
+			"Hello World",
+		},
+		{
+			"int",
+			"int 42",
+			int(42),
+		},
+		{
+			"uint",
+			"uint 43",
+			uint(43),
+		},
+		{
+			"float",
+			"float 44",
+			float32(44),
+		},
+		{
+			"time",
+			"time 05-18-1999_6:42pm",
+			tim,
+		},
+		{
+			"duration",
+			"dur 4h2m",
+			dur,
+		},
+		{
+			"string int",
+			"string Lmao int 42",
+			&T{
+				String: "Lmao",
+				Int:    42,
+			},
+		},
+		{
+			"string time",
+			"string What's Up Boy time 05-18-1999_6:42pm",
+			&T{
+				String: "What's Up Boy",
+				Time:   tim,
+			},
+		},
+	}
+
+	for _, c := range cases {
+		b.Run(c.name, func(b *testing.B) {
+			nl.P(c.expression)
+		})
+	}
+}