From 394bf609a81e4fbfb0bddb50a9dccdddf14d69f8 Mon Sep 17 00:00:00 2001
From: Shrey Desai <shreydesai@me.com>
Date: Thu, 12 May 2016 21:56:57 -0700
Subject: [PATCH 1/2] added tests file

---
 tests.py | 0
 1 file changed, 0 insertions(+), 0 deletions(-)
 create mode 100644 tests.py

diff --git a/tests.py b/tests.py
new file mode 100644
index 0000000..e69de29

From beb60cde77e9b39b1349847c34552064f1657a07 Mon Sep 17 00:00:00 2001
From: Shrey Desai <shreydesai@me.com>
Date: Thu, 12 May 2016 22:33:38 -0700
Subject: [PATCH 2/2] added test case for key methods

---
 tests.py | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/tests.py b/tests.py
index e69de29..0768e89 100644
--- a/tests.py
+++ b/tests.py
@@ -0,0 +1,41 @@
+from unittest import TestCase, main
+import pickledb
+
+class TestKeyMethods(TestCase):
+    def setUp(self):
+        self.db = pickledb.load("test.db", False)
+
+    def test_set_string(self):
+        self.db.set("key", "string")
+        value = self.db.get("key")
+        self.assertEqual(value, "string")
+
+    def test_set_int(self):
+        self.db.set("key", 1)
+        value = self.db.get("key")
+        self.assertEqual(value, 1)
+
+    def test_set_double(self):
+        self.db.set("key", 1.0)
+        value = self.db.get("key")
+        self.assertEqual(value, 1.0)
+
+    def test_set_bool(self):
+        self.db.set("key", True)
+        value = self.db.get("key")
+        self.assertEqual(value, True)
+
+    def test_getall(self):
+        self.db.set("key1", "value1")
+        self.db.set("key2", "value2")
+        self.db.set("key3", "value3")
+        for k in ["key1", "key2", "key3"]:
+            self.assertTrue(k in self.db.getall())
+
+    def test_rem(self):
+        self.db.set("key", "value")
+        self.db.rem("key")
+        self.assertTrue("key" not in self.db.getall())
+
+if __name__ == "__main__":
+    main()