[artiq] cleanup module functionality tests

This commit is contained in:
abdul124 2025-01-17 17:56:43 +08:00
parent 2d275949b8
commit f817d3347b
3 changed files with 33 additions and 24 deletions

26
nac3artiq/demo/module.py Normal file
View File

@ -0,0 +1,26 @@
from min_artiq import *
from numpy import int32
# Global Variable Definition
X: Kernel[int32] = 1
# TopLevelFunction Defintion
@kernel
def display_X():
print_int32(X)
# TopLevel Class Definition
@nac3
class A:
@kernel
def __init__(self):
self.set_x(1)
@kernel
def set_x(self, new_val: int32):
global X
X = new_val
@kernel
def get_X(self) -> int32:
return X

View File

@ -1,7 +1,5 @@
from min_artiq import *
import tests.string_attribute_issue337 as issue337
import tests.support_class_attr_issue102 as issue102
import tests.global_variables as global_variables
import module as module_definition
@nac3
class TestModuleSupport:
@ -13,17 +11,16 @@ class TestModuleSupport:
@kernel
def run(self):
# Accessing classes
issue337.Demo().run()
obj = issue102.Demo()
obj.attr3 = 3
obj = module_definition.A()
obj.get_X()
obj.set_x(2)
# Calling functions
global_variables.inc_X()
global_variables.display_X()
module_definition.display_X()
# Updating global variables
global_variables.X = 9
global_variables.display_X()
module_definition.X = 9
module_definition.display_X()
if __name__ == "__main__":
TestModuleSupport().run()

View File

@ -1,14 +0,0 @@
from min_artiq import *
from numpy import int32
X: Kernel[int32] = 1
@rpc
def display_X():
print_int32(X)
@kernel
def inc_X():
global X
X += 1