Fixing unit tests for the payload format.

Author: jirioc

backends/nxp/tests/test_neutron_backend.py

@@ -28,23 +28,29 @@ def test_neutron_backend__single_conv_model__payload_header_channels_last():
     edge_program_manager = to_quantized_edge_program(Conv2dModule(bias=False), (1, 4, 32, 32))
     payload = edge_program_manager.exported_program().graph_module.lowered_module_0.processed_bytes
 
-    assert payload[0] == 0x1  # Single input
-    assert payload[1] == 0x1  # Single output
-    assert payload[2] == 0x1  # Channels last
-    assert payload[3] == 0x1  # Channels last
-    assert all(byte == 0x0 for byte in payload[4:16])  # Aligned to 16 bytes
+    assert payload[0] == 0x1  # Number of Neutron node inputs
+    assert payload[1] == 0x1  # Number of Neutron node outputs
+    assert payload[2] == 0x1  # Number of model inputs
+    assert payload[3] == 0x1  # Channels last 0-th Neutron input
+    assert payload[4] == 0x1  # Channels last 0-th Neutron output
+    assert payload[5] == 0x0  # Map 0-th Neutron input to 0-th model input
+    assert payload[6] == 0x0  # Map 0-th Neutron output to 0-th model output
+    assert all(byte == 0x0 for byte in payload[7:16])  # Aligned to 16 bytes
     assert payload[17] != 0x0  # Followed by non-zero content
 
 
 def test_neutron_backend__linear_softmax_model__payload_header_formatless():
     edge_program_manager = to_quantized_edge_program(LinearSoftmaxModule(), (1, 12))
     payload = edge_program_manager.exported_program().graph_module.lowered_module_0.processed_bytes
 
-    assert payload[0] == 0x1  # Single input
-    assert payload[1] == 0x1  # Single output
-    assert payload[2] == 0x0  # Formatless
-    assert payload[3] == 0x0  # Formatless
-    assert all(byte == 0x0 for byte in payload[4:16])  # Aligned to 16 bytes
+    assert payload[0] == 0x1  # Number of Neutron node inputs
+    assert payload[1] == 0x1  # Number of Neutron node outputs
+    assert payload[2] == 0x1  # Number of model inputs
+    assert payload[3] == 0x0  # Formatless 0-th Neutron input
+    assert payload[4] == 0x0  # Formatless 0-th Neutron output
+    assert payload[5] == 0x0  # Map 0-th Neutron input to 0-th model input
+    assert payload[6] == 0x0  # Map 0-th Neutron output to 0-th model output
+    assert all(byte == 0x0 for byte in payload[7:16])  # Aligned to 16 bytes
     assert payload[17] != 0x0  # Followed by non-zero content