Source code for sqlobject.tests.test_aggregates

from sqlobject import FloatCol, IntCol, SQLObject
from sqlobject.tests.dbtest import setupClass


# Test MIN, AVG, MAX, COUNT, SUM


[docs]class IntAccumulator(SQLObject): so_value = IntCol()
[docs]class FloatAccumulator(SQLObject): so_value = FloatCol()
[docs]def test_integer(): setupClass(IntAccumulator) IntAccumulator(so_value=1) IntAccumulator(so_value=2) IntAccumulator(so_value=3) assert IntAccumulator.select().min(IntAccumulator.q.so_value) == 1 assert IntAccumulator.select().avg(IntAccumulator.q.so_value) == 2 assert IntAccumulator.select().max(IntAccumulator.q.so_value) == 3 assert IntAccumulator.select().sum(IntAccumulator.q.so_value) == 6 assert IntAccumulator.select(IntAccumulator.q.so_value > 1).\ max(IntAccumulator.q.so_value) == 3 assert IntAccumulator.select(IntAccumulator.q.so_value > 1).\ sum(IntAccumulator.q.so_value) == 5
[docs]def floatcmp(f1, f2): if abs(f1 - f2) < 0.1: return 0 if f1 < f2: return 1 return -1
[docs]def test_float(): setupClass(FloatAccumulator) FloatAccumulator(so_value=1.2) FloatAccumulator(so_value=2.4) FloatAccumulator(so_value=3.8) assert floatcmp( FloatAccumulator.select().min(FloatAccumulator.q.so_value), 1.2) == 0 assert floatcmp( FloatAccumulator.select().avg(FloatAccumulator.q.so_value), 2.5) == 0 assert floatcmp( FloatAccumulator.select().max(FloatAccumulator.q.so_value), 3.8) == 0 assert floatcmp( FloatAccumulator.select().sum(FloatAccumulator.q.so_value), 7.4) == 0
[docs]def test_many(): setupClass(IntAccumulator) IntAccumulator(so_value=1) IntAccumulator(so_value=1) IntAccumulator(so_value=2) IntAccumulator(so_value=2) IntAccumulator(so_value=3) IntAccumulator(so_value=3) attribute = IntAccumulator.q.so_value assert list(IntAccumulator.select().accumulateMany( ("MIN", attribute), ("AVG", attribute), ("MAX", attribute), ("COUNT", attribute), ("SUM", attribute) )) == [1, 2, 3, 6, 12] assert list(IntAccumulator.select(distinct=True).accumulateMany( ("MIN", attribute), ("AVG", attribute), ("MAX", attribute), ("COUNT", attribute), ("SUM", attribute) )) == [1, 2, 3, 3, 6]