Various fixes; start implementing tests

5 files changed, 185 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..71bd3ac
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,18 @@
+import datetime as dt
+
+import pytest
+
+from blc2.workspace import Workspace 
+from blc2.topology import Fixture
+
+@pytest.fixture
+def ws():
+    """Return a workspace."""
+    return Workspace("", "" ,"", dt.datetime.now())
+
+@pytest.fixture
+def aws():
+    """Return a more advanced workspace with 4 channels on 1 fixture."""
+    w = Workspace("", "", "", dt.datetime.now())
+    Fixture(w, id_=0, channel_count=4)
+    return w
diff --git a/tests/silence.m4a b/tests/silence.m4a
new file mode 100644
index 0000000..0775df6
--- /dev/null
+++ b/tests/silence.m4a
diff --git a/tests/test_functions_audio.py b/tests/test_functions_audio.py
new file mode 100644
index 0000000..de6aa2a
--- /dev/null
+++ b/tests/test_functions_audio.py
@@ -0,0 +1,36 @@
+import os
+
+from blc2.functions.audio import Audio
+
+def test_audio(aws):
+    a = Audio(aws)
+    
+    a.fade_out = 1000
+    a.fade_in = 1000
+    a.filename = "nonexistant"
+    
+    assert a.audio_scope == {"nonexistant"}
+    assert not a.scope
+    assert a.duration == 0
+    assert a.actual_duration == 0
+
+    a.filename = "tests/silence.m4a"
+    assert a.audio_scope == {"tests/silence.m4a"}
+    assert a.duration == 2024
+    assert a.actual_duration == 3024
+
+    a.fade_out = 500
+    assert a.fade_out == 500
+    assert a.duration == 2524
+    assert a.actual_duration == 3024
+
+    lc, ac, _ = a.render(0)
+    assert not lc
+    assert len(ac) == 1
+    assert ac[0][1:] == ("tests/silence.m4a", 0, 1000, 2524, 500)
+
+    _, ac2, _ = a.render(3000)
+    assert ac2 == ac
+
+    _, ac, _ = a.render(5000)
+    assert not ac
diff --git a/tests/test_functions_scene.py b/tests/test_functions_scene.py
new file mode 100644
index 0000000..7a5b78b
--- /dev/null
+++ b/tests/test_functions_scene.py
@@ -0,0 +1,76 @@
+import pytest
+
+from blc2.functions.scene import Scene
+from blc2.constants import INFTY
+
+def test_scene(aws):
+    """Test creation and modification of scenes."""
+    c1, c2, c3, c4 = aws.fixtures[0].channels
+
+    s1 = Scene(aws)
+    assert s1.scope == frozenset()
+    assert s1.scope == frozenset()
+
+    values = {c1: 0, c2: 1, c3: 2, c4: 4}
+    s1.values = values
+    assert s1.values == values
+    assert s1.scope == frozenset(values.keys())
+
+    s1.update({c1: 10})
+    assert s1.values[c1] == 10
+    assert s1.scope == frozenset(values.keys())
+    for c, v in values.items():
+        if c == c1:
+            continue
+        assert v == s1.values[c]
+
+    s1.set(values)
+    assert s1.values == values
+    assert s1.scope == frozenset(values.keys())
+
+    del s1[c1]
+    assert s1.scope == {c2, c3, c4}
+    assert c1 not in s1.values
+    assert s1[c1] is None
+
+    s1[c1] = 10
+    assert s1.scope == frozenset(values.keys())
+    assert s1[c1] == 10
+
+    assert s1.duration == INFTY
+    assert s1.actual_duration == INFTY
+    assert s1.fade_in == 0
+    assert s1.fade_out == 0
+    assert not s1.audio_scope
+
+    with pytest.raises(Exception):
+        s1.duration = 0
+
+    with pytest.raises(Exception):
+        s1.actual_duration = 0
+
+    with pytest.raises(Exception):
+        s1.fade_in = 0
+
+    with pytest.raises(Exception):
+        s1.fade_out = 0
+
+    with pytest.raises(Exception):
+        s1.scope = frozenset()
+
+    with pytest.raises(Exception):
+        s1.audio_scope = frozenset()
+
+    s1.set(values)
+    lc, ac, _ = s1.render(0)
+    assert ac == ()
+    assert dict(lc) == values
+
+    lc, ac, _ = s1.render(1234567890)
+    assert ac == ()
+    assert dict(lc) == values
+
+    s1[c1] = 10
+    lc, ac, _ = s1.render(1)
+    assert dict(lc)[c1] == 10
+    assert False not in (v == values[c] for c, v in lc if c != c1)
diff --git a/tests/test_topology.py b/tests/test_topology.py
new file mode 100644
index 0000000..0579d63
--- /dev/null
+++ b/tests/test_topology.py
@@ -0,0 +1,55 @@
+"""Module for testing topology behaviour."""
+
+import pytest
+
+from blc2.topology import Fixture
+from blc2.workspace import Workspace
+
+def test_fixture_create(ws):
+    """Test basic fixture creation."""
+    f = Fixture(ws, id_=0, name="Test 1")
+    assert f.id in ws.fixtures
+    assert f.id == 0
+    assert f.name == "Test 1"
+    assert not f.channels
+
+    with pytest.raises(ValueError):
+        Fixture(ws, id_=0, name="Test 2")
+
+    assert ws.fixtures[0] == f
+    assert len(ws.fixtures) == 1
+
+    f2 = Fixture(ws, name="Test 3")
+    assert f2.id is not None
+    assert ws.fixtures[f2.id] == f2
+    assert len(ws.fixtures) == 2
+
+    f3 = Fixture(ws, channel_count = 3)
+    assert len(f3.channels) == 3
+
+def test_fixture_channels(ws):
+    """Test fixture channel manipulations."""
+    f = Fixture(ws, channel_count=3)
+    for i in range(3):
+        f.channels[i].address = (0, i)
+    channels = tuple(f.channels)
+
+    with pytest.raises(ValueError):
+        f.channel_count = -1
+    assert len(f.channels) == 3
+
+    ## Verify that the channels haven't been changed
+    for a, b in zip(f.channels, channels):
+        assert id(a) == id(b)
+    
+    f.channel_count = 2
+    assert len(f.channels) == 2
+    for a, b in zip(f.channels, channels[:2]):
+        assert id(a) == id(b)
+    
+    f.channel_count = 3
+    assert len(f.channels) == 3
+    for a, b in zip(f.channels, channels[:2]):
+        assert id(a) == id(b)
+
+    assert id(f.channels[2]) != id(channels[2])