python-FloatImg/tests/dumps.py

"""test of file dump functions """
import os

import floatimg

TEST_PATH = os.path.join(os.path.dirname(__file__), "test.dump")

# TODO refactor to fixture
def create_dump():
    width, height = 5, 5
    img = floatimg.create_rgb(width, height)
    # fill it to have something to compare
    img.fill((64.0, 64.0, 64.0))
    img.dump(TEST_PATH)


def test_dump_to_file():
    create_dump()

    assert os.path.exists(TEST_PATH)
    assert os.path.isfile(TEST_PATH)
    assert os.path.getsize(TEST_PATH) == 320

    os.remove(TEST_PATH)


def test_infos():
    create_dump()
    floatimg.fileinfos(TEST_PATH) == 5, 5, floatimg.RGB
    os.remove(TEST_PATH)


def test_load():
    create_dump()
    width, height = 5, 5
    img = floatimg.create_rgb(width, height)

    img.load(TEST_PATH)
    for y in range(height):
        for x in range(width):
            assert img.get(x, y) == [64.0, 64.0, 64.0]
    os.remove(TEST_PATH)


def test_create():
    create_dump()
    img = floatimg.create_from_dump(TEST_PATH)
    for y in range(img.height):
        for x in range(img.width):
            assert img.get(x, y) == [64.0, 64.0, 64.0]
    os.remove(TEST_PATH)