From 571cbe07a045b174213369aa3f0514096cfce4c7 Mon Sep 17 00:00:00 2001
From: Worros <carl.gherardi@gmail.com>
Date: Thu, 19 Nov 2009 15:28:48 +0800
Subject: [PATCH] Add test functions for SQLite

Added test for VARIANCE()

Added test for floor()
---
 pyfpdb/test_Database.py | 51 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 pyfpdb/test_Database.py

diff --git a/pyfpdb/test_Database.py b/pyfpdb/test_Database.py
new file mode 100644
index 00000000..9283f46d
--- /dev/null
+++ b/pyfpdb/test_Database.py
@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+import sqlite3
+import fpdb_db
+import math
+
+# Should probably use our wrapper classes - creating sqlite db in memory
+sqlite3.register_converter("bool", lambda x: bool(int(x)))
+sqlite3.register_adapter(bool, lambda x: "1" if x else "0")
+
+con = sqlite3.connect(":memory:")
+con.isolation_level = None
+
+#Floor function
+con.create_function("floor", 1, math.floor)
+
+#Mod function
+tmp = fpdb_db.sqlitemath()
+con.create_function("mod", 2, tmp.mod)
+
+# Aggregate function VARIANCE()
+con.create_aggregate("variance", 1, fpdb_db.VARIANCE)
+
+
+cur = con.cursor()
+
+def testSQLiteVarianceFunction():
+    cur.execute("CREATE TABLE test(i)")
+    cur.execute("INSERT INTO test(i) values (1)")
+    cur.execute("INSERT INTO test(i) values (2)")
+    cur.execute("INSERT INTO test(i) values (3)")
+    cur.execute("SELECT variance(i) from test")
+    result = cur.fetchone()[0]
+
+    print "DEBUG: Testing variance function"
+    print "DEBUG: result: %s expecting: 0.666666 (result-expecting ~= 0.0): %s" % (result, (result - 0.66666))
+    cur.execute("DROP TABLE test")
+    assert (result - 0.66666) <= 0.0001
+
+def testSQLiteFloorFunction():
+    vars    = [0.1, 1.5, 2.6, 3.5, 4.9]
+    cur.execute("CREATE TABLE test(i float)")
+    for var in vars:
+        cur.execute("INSERT INTO test(i) values(%f)" % var)
+    cur.execute("SELECT floor(i) from test")
+    result = cur.fetchall()
+    print "DEBUG: result: %s" % result
+    answer = 0
+    for i in result:
+        print "DEBUG: int(var): %s" % int(i[0])
+        assert answer == int(i[0])
+        answer = answer + 1