#

import pytest
import sqlitent
import collections
import datetime
from random import random

#

Fixtures

#

@pytest.fixture
def empty():
    return sqlitent.sqlitent(':memory:')


@pytest.fixture
def point():
    return collections.namedtuple('point', ['x', 'y', 'z'])


@pytest.fixture
def timestamp():
    return collections.namedtuple('timestamp', ['dt'])


@pytest.fixture
def nonempty(empty, point):
    for _ in range(25):
        p = point(random(), random(), random())
        assert empty.add(p) == None

    return empty

#

Testing basic operations

#

def test_empty(empty):
    assert not len(empty)
    assert list(empty) == []

#

def test_nonemtpy(nonempty):
    assert len(nonempty)
    assert list(nonempty)

#

def test_add(empty, point):
    num_entries = len(empty)

    ps = [
        point(random(), random(), random()),
        point(    None, random(), random()),
        point(random(),     None, random()),
        point(random(), random(),     None)
    ]

    for p in ps:
        num_entries = len(empty)
        assert empty.add(p) == None
        assert len(empty) == num_entries + 1
        assert p in empty

#

def test_remove(nonempty):
    for p in nonempty:
        num_entries = len(nonempty)
        assert p in nonempty
        assert nonempty.remove(p) == None
        assert len(nonempty) == num_entries + -1
        assert p not in nonempty

    assert not len(nonempty)
    assert list(nonempty) == []

#

Testing bulk operations

#

def test_insert(empty, point):
    num_entries = len(empty)
    assert empty.insert([
        point(random(), random(), random()),
        point(    None, random(), random()),
        point(random(),     None, random()),
        point(random(), random(),     None)
    ]) == None
    assert len(empty) == num_entries + 4

#

def test_delete(nonempty):
    assert nonempty.delete(list(nonempty)) == None
    assert len(nonempty) == 0

#

Testing operations on complex types

#

def test_add_datetime(empty, timestamp):
    ts = timestamp(datetime.datetime.now())
    assert empty.add(ts) == None
    assert ts in empty

#

def test_remove_datetime(nonempty, timestamp):
    ts = timestamp(datetime.datetime.now())
    assert nonempty.add(ts) == None
    assert ts in nonempty
    xx = nonempty.one(timestamp)
    assert isinstance(xx, timestamp)
    assert xx == ts
    assert nonempty.remove(xx) == None
    assert ts not in nonempty

test_behaviour.py

Fixtures

Testing basic operations

Testing bulk operations

Testing operations on complex types