From ff664666b8615705f7538823fed310007f88f214 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Bourdeauducq?= <sb@m-labs.hk>
Date: Mon, 19 Aug 2024 22:21:08 +0800
Subject: [PATCH] test/stress: partial port to NAC3

---
 artiq/test/coredevice/test_stress.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/artiq/test/coredevice/test_stress.py b/artiq/test/coredevice/test_stress.py
index 514a2d8ac..4f2183a41 100644
--- a/artiq/test/coredevice/test_stress.py
+++ b/artiq/test/coredevice/test_stress.py
@@ -1,23 +1,27 @@
-import os
-import time
 import unittest
 
+from numpy import int32
+
 from artiq.experiment import *
 from artiq.test.hardware_testbench import ExperimentCase
+from artiq.coredevice.core import Core
 
-
+@nac3
 class _Stress(EnvExperiment):
+    core: KernelInvariant[Core]
+
     def build(self):
         self.setattr_device("core")
 
-    @rpc(flags={"async"})
-    def sink(self, data):
+    # NAC3TODO https://git.m-labs.hk/M-Labs/nac3/issues/182
+    @rpc #(flags={"async"})
+    def sink(self, data: int32):
         pass
 
     @kernel
-    def async_rpc(self, n):
+    def async_rpc(self, n: int32):
         for _ in range(n):
-            self.sink(b"")
+            self.sink(0)
 
 
 class StressTest(ExperimentCase):