From 74c14b3add928fe92de85832e89db4b51147f764 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 13 Jul 2022 15:55:43 +0100
Subject: [PATCH] [tests] Update tests to allow Alveo build tests and add
marker to buildflow test
---
tests/end2end/test_end2end_bnn_pynq.py | 15 ---------------
tests/util/test_build_dataflow.py | 1 +
2 files changed, 1 insertion(+), 15 deletions(-)
diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py
index ab82a00c2..4139c74de 100644
--- a/tests/end2end/test_end2end_bnn_pynq.py
+++ b/tests/end2end/test_end2end_bnn_pynq.py
@@ -673,9 +673,6 @@ class TestEnd2End:
@pytest.mark.vitis
@pytest.mark.parametrize("kind", ["zynq", "alveo"])
def test_build(self, topology, wbits, abits, QONNX_export, kind):
- # temporarily adding skip for alveo builds
- if kind == "alveo":
- pytest.skip("Alveo tests temporarily excluded")
if kind == "alveo" and ("VITIS_PATH" not in os.environ):
pytest.skip("VITIS_PATH not set")
prev_chkpt_name = get_checkpoint_name(
@@ -698,9 +695,6 @@ class TestEnd2End:
@pytest.mark.vitis
@pytest.mark.parametrize("kind", ["zynq", "alveo"])
def test_make_pynq_driver(self, topology, wbits, abits, QONNX_export, kind):
- # temporarily adding skip for alveo builds
- if kind == "alveo":
- pytest.skip("Alveo tests temporarily excluded")
if kind == "alveo" and ("VITIS_PATH" not in os.environ):
pytest.skip("VITIS_PATH not set")
prev_chkpt_name = get_checkpoint_name(
@@ -715,9 +709,6 @@ class TestEnd2End:
@pytest.mark.parametrize("kind", ["zynq", "alveo"])
def test_deploy(self, topology, wbits, abits, QONNX_export, kind):
- # temporarily adding skip for alveo builds
- if kind == "alveo":
- pytest.skip("Alveo tests temporarily excluded")
prev_chkpt_name = get_checkpoint_name(
topology, wbits, abits, QONNX_export, "driver_" + kind
)
@@ -741,9 +732,6 @@ class TestEnd2End:
@pytest.mark.parametrize("kind", ["zynq", "alveo"])
def test_run_on_hw(self, topology, wbits, abits, QONNX_export, kind):
- # temporarily adding skip for alveo builds
- if kind == "alveo":
- pytest.skip("Alveo tests temporarily excluded")
prev_chkpt_name = get_checkpoint_name(
topology, wbits, abits, QONNX_export, "deploy_" + kind
)
@@ -768,9 +756,6 @@ class TestEnd2End:
@pytest.mark.parametrize("kind", ["zynq", "alveo"])
def test_throughput_hw(self, topology, wbits, abits, QONNX_export, kind):
- # temporarily adding skip for alveo builds
- if kind == "alveo":
- pytest.skip("Alveo tests temporarily excluded")
prev_chkpt_name = get_checkpoint_name(
topology, wbits, abits, QONNX_export, "deploy_" + kind
)
diff --git a/tests/util/test_build_dataflow.py b/tests/util/test_build_dataflow.py
index d33a4f2fd..cdf69aebd 100644
--- a/tests/util/test_build_dataflow.py
+++ b/tests/util/test_build_dataflow.py
@@ -39,6 +39,7 @@ from finn.util.basic import make_build_dir
@pytest.mark.slow
@pytest.mark.vivado
+@pytest.mark.end2end
def test_end2end_build_dataflow_directory():
test_dir = make_build_dir("test_build_dataflow_directory_")
target_dir = test_dir + "/build_dataflow"
--
GitLab