diff --git a/elasticai/creator/nn/fixed_point/conv1d/layer/conv1d_test.py b/elasticai/creator/nn/fixed_point/conv1d/layer/conv1d_test.py index 28756abf..a8e1c9b9 100644 --- a/elasticai/creator/nn/fixed_point/conv1d/layer/conv1d_test.py +++ b/elasticai/creator/nn/fixed_point/conv1d/layer/conv1d_test.py @@ -1,6 +1,10 @@ +from typing import cast + import pytest import torch +from elasticai.creator.file_generation.in_memory_path import InMemoryFile, InMemoryPath + from .conv1d import Conv1d @@ -135,3 +139,19 @@ def test_bias_addition() -> None: inputs = torch.tensor([0.0, 0.0]) predictions = conv(inputs) assert predictions.tolist() == [0.5] + + +def test_conv1d_layer_creates_correct_design(conv1d: Conv1d) -> None: + expected_conv1d_code = """-- Dummy File for testing implementation of conv1d Design +16 +8 +3 +4 +2""" + + design = conv1d.translate("conv1d") + destination = InMemoryPath("conv1d", parent=None) + design.save_to(destination) + actual_conv1d_code = "\n".join(cast(InMemoryFile, destination["conv1d"]).text) + + assert expected_conv1d_code == actual_conv1d_code