Skip to content

Commit

Permalink
feat(tests): add small integration test to verify that conv1d layer g…
Browse files Browse the repository at this point in the history
…enerates correct design
  • Loading branch information
julianhoever committed Sep 13, 2023
1 parent d9482ce commit da45cc3
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions elasticai/creator/nn/fixed_point/conv1d/layer/conv1d_test.py
Original file line number Diff line number Diff line change
@@ -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


Expand Down Expand Up @@ -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

0 comments on commit da45cc3

Please sign in to comment.