added input/output node tests

.github/workflows/build.yml

@@ -30,7 +30,7 @@ jobs:
           # stop the build if there are Python syntax errors or undefined names
           ruff --format=github --select=E9,F63,F7,F82 --target-version=py37 .
           # default set of ruff rules with GitHub Annotations
-          ruff --format=github --target-version=py37 .
+          ruff --format=github --target-version=py37 --exclude=docs/ .
       - name: Test with pytest
         run: |
           pytest

nir/test/test_ir.py

@@ -6,3 +6,16 @@ def test_simple():
     assert ir.nodes[0].weights == [1, 2, 3]
     assert ir.nodes[0].bias == 4
     assert ir.edges == [(0, 0)]
+
+def test_simple_with_input_output():
+    ir = nir.NIR(
+        nodes=[
+            nir.Input(shape=[3,]),
+            nir.Linear(weights=[1, 2, 3], bias=4),
+            nir.Output()],
+        edges=[(0, 1), (1,2)])
+    assert ir.nodes[0].shape == [3,]
+    assert ir.nodes[1].weights == [1, 2, 3]
+    assert ir.nodes[1].bias == 4
+    assert ir.edges == [(0, 1), (1, 2)]
+

nir/test/test_readwrite.py

@@ -36,3 +36,12 @@ def test_leaky_integrator_and_fire():
         edges=[(0, 0)],
     )
     factory_test_graph(ir)
+
+def test_simple_with_read_write():
+    ir = nir.NIR(
+        nodes=[nir.Input(shape=[3,]),
+               nir.Linear(weights=[1, 2, 3], bias=4),
+               nir.Output()],
+        edges=[(0, 1), (1,2)]
+    )
+    factory_test_graph(ir)