From 0d26e737467facf2f33bac5f90ae1a4583fc40f4 Mon Sep 17 00:00:00 2001
From: edwardro22 <edwardro22@gmail.com>
Date: Tue, 4 Jul 2017 23:22:15 +0300
Subject: [PATCH] added tests for Contains

---
 utils/dateseries_test.go | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/utils/dateseries_test.go b/utils/dateseries_test.go
index 225897124..6cfe92331 100644
--- a/utils/dateseries_test.go
+++ b/utils/dateseries_test.go
@@ -283,3 +283,39 @@ func TestDateseriesYearsParse(t *testing.T) {
 		t.Errorf("Expected: %+v, received: %+v", ys2, ys1)
 	}
 }
+
+func TestDateseriesWeekDaysContains(t *testing.T) {
+	wds := WeekDays{time.Monday, time.Tuesday}
+	if wds.Contains(time.Monday) != true {
+		t.Errorf("Expected: true, got: %v", !true)
+	} else if wds.Contains(time.Wednesday) != false {
+		t.Errorf("Expected: false, got: %v", !false)
+	}
+}
+
+func TestDateseriesMonthsContains(t *testing.T) {
+	m := Months{5, 6, 7, 8}
+	if m.Contains(5) != true {
+		t.Errorf("Expected: true, got: %v", !true)
+	} else if m.Contains(4) != false {
+		t.Errorf("Expected: false, got: %v", !false)
+	}
+}
+
+func TestDateseriesMonthDaysContains(t *testing.T) {
+	md := MonthDays{24, 25, 26}
+	if md.Contains(24) != true {
+		t.Errorf("Expected: true, got: %v", !true)
+	} else if md.Contains(23) != false {
+		t.Errorf("Expected: false, got: %v", !false)
+	}
+}
+
+func TestDateseriesYearsContains(t *testing.T) {
+	ys := Years{2013, 2014, 2015}
+	if ys.Contains(2013) != true {
+		t.Errorf("Expected: true, got: %v", !true)
+	} else if ys.Contains(2012) != false {
+		t.Errorf("Expected: false, got: %v", !false)
+	}
+}