ONE - On-device Neural Engine
Loading...
Searching...
No Matches
test_utils.TestRunner Class Reference

Public Member Functions

 __init__ (self, str output_dir)
 
 register (self, TestCase test_case)
 
 run (self)
 

Data Fields

 test_cases
 
 output_dir
 

Detailed Description

Definition at line 80 of file test_utils.py.

Constructor & Destructor Documentation

◆ __init__()

test_utils.TestRunner.__init__ (   self,
str  output_dir 
)

Definition at line 81 of file test_utils.py.

81 def __init__(self, output_dir: str):
82 self.test_cases = list()
83 self.output_dir = output_dir
84

Member Function Documentation

◆ register()

test_utils.TestRunner.register (   self,
TestCase  test_case 
)

Definition at line 85 of file test_utils.py.

85 def register(self, test_case: TestCase):
86 self.test_cases.append(test_case)
87

References test_utils.TestRunner.test_cases.

◆ run()

test_utils.TestRunner.run (   self)

Definition at line 88 of file test_utils.py.

88 def run(self):
89 for test_case in self.test_cases:
90 print("Generate test case: " + test_case.name)
91 _dump_npy_included_json(self.output_dir + '/' + test_case.name,
92 test_case.generate())
void run(std::ofstream &os, const circle::Model *model)

References test_utils._dump_npy_included_json(), test_utils.TestRunner.output_dir, and test_utils.TestRunner.test_cases.

Referenced by package.infer.session.inference().

Field Documentation

◆ output_dir

test_utils.TestRunner.output_dir

Definition at line 83 of file test_utils.py.

Referenced by test_utils.TestRunner.run().

◆ test_cases

test_utils.TestRunner.test_cases

Definition at line 82 of file test_utils.py.

Referenced by test_utils.TestRunner.register(), and test_utils.TestRunner.run().


The documentation for this class was generated from the following file: