test_simplesimdb.py

import os.path

import simplesimdb as sim

# Run with pytest -s . to see stdout output


def test_construction_and_destruction():
    print("TEST")
    m = sim.Manager()
    assert os.path.isdir("data")
    m.delete_all()
    assert not os.path.isdir("data")


def test_creation():
    print("TEST CREATION")
    m = sim.Manager(directory="creation_test", executable="cp", filetype="json")
    assert m.directory == "creation_test"
    assert m.executable == "cp"
    assert m.filetype == "json"
    inputdata = {"Hello": "World"}
    m.create(inputdata)
    content = m.table()
    assert content == [inputdata]
    m.delete_all()
    assert not os.path.isdir(m.directory)


def test_selection():
    print("TEST SELECTION")
    m = sim.Manager(directory="selection_test", executable="cp", filetype="json")
    inputdata = {"Hello": "World"}
    inputdata2 = {"Hello": "World!"}
    m.create(inputdata)
    m.create(inputdata2)
    data = m.select(inputdata)
    assert os.path.isfile(data)
    assert data == m.outfile(inputdata)
    m.delete_all()
    assert not os.path.isdir(m.directory)


def test_restart():
    print("TEST RESTART")
    m = sim.Manager(directory="restart_test", executable="touch", filetype="th")
    inputdata = {"Hello": "World"}
    for i in range(0, 17):
        m.create(inputdata, i)
    count = m.count(inputdata)
    assert count == 17
    data = m.select(inputdata, 3)
    assert os.path.isfile(data)
    inputdata2 = {"Hello2": "World"}
    for i in range(0, 7):
        m.create(inputdata2, i)
    content = m.table()
    # Check ordered content
    assert content == [inputdata2, inputdata]
    files = m.files()
    assert len(files) == 24
    m.delete_all()
    assert not os.path.isdir(m.directory)


def test_named_creation():
    print("TEST NAMED CREATION")
    m = sim.Manager(
        directory="creation_named_test", executable="touch", filetype="json"
    )
    inputdata = {"Hello": "World"}
    m.create(inputdata, 0)
    m.delete(inputdata, 0)
    m.create(inputdata, 0, "hello")
    m.create(inputdata, 1, "hello")
    content = m.table()
    print(m.files())
    assert content == [inputdata]
    m.delete_all()
    assert not os.path.isdir(m.directory)


def test_repeater():
    print("TEST REPEATER")
    m = sim.Repeater("touch", "temp.json", "temp.nc")
    inputdata = {"Hello": "World"}
    m.run(inputdata, error="display", stdout="display")
    assert os.path.isfile("temp.json")
    assert os.path.isfile("temp.nc")
    m.executable = "echo"
    m.run(inputdata, error="display", stdout="display")
    m.clean()
    assert not os.path.isfile("temp.json")
    assert not os.path.isfile("temp.nc")