From da45cc30b481ee27e9bd9d5d172e29a0bd0b519f Mon Sep 17 00:00:00 2001 From: Julian Hoever Date: Wed, 13 Sep 2023 20:19:48 +0200 Subject: [PATCH] feat(tests): add small integration test to verify that conv1d layer generates correct design --- .../fixed_point/conv1d/layer/conv1d_test.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) 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