Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
183 changes: 183 additions & 0 deletions Lib/test/test_generated_cases.py
Original file line number Diff line number Diff line change
Expand Up @@ -2069,6 +2069,189 @@ def test_missing_override_failure(self):
with self.assertRaisesRegex(AssertionError, "All abstract uops"):
self.run_cases_test(input, input2, output)

def test_validate_uop_input_length_mismatch(self):
input = """
op(OP, (arg1 -- out)) {
SPAM();
}
"""
input2 = """
op(OP, (arg1, arg2 -- out)) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Must have the same number of inputs"):
self.run_cases_test(input, input2, output)

def test_validate_uop_output_length_mismatch(self):
input = """
op(OP, (arg1 -- out)) {
SPAM();
}
"""
input2 = """
op(OP, (arg1 -- out1, out2)) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Must have the same number of outputs"):
self.run_cases_test(input, input2, output)

def test_validate_uop_input_name_mismatch(self):
input = """
op(OP, (foo -- out)) {
SPAM();
}
"""
input2 = """
op(OP, (bar -- out)) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Inputs must have equal names"):
self.run_cases_test(input, input2, output)

def test_validate_uop_output_name_mismatch(self):
input = """
op(OP, (arg1 -- foo)) {
SPAM();
}
"""
input2 = """
op(OP, (arg1 -- bar)) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Outputs must have equal names"):
self.run_cases_test(input, input2, output)

def test_validate_uop_unused_input(self):
input = """
op(OP, (unused -- )) {
}
"""
input2 = """
op(OP, (foo -- )) {
}
"""
output = """
case OP: {
stack_pointer += -1;
assert(WITHIN_STACK_BOUNDS());
break;
}
"""
self.run_cases_test(input, input2, output)

input = """
op(OP, (foo -- )) {
}
"""
input2 = """
op(OP, (unused -- )) {
}
"""
output = """
case OP: {
stack_pointer += -1;
assert(WITHIN_STACK_BOUNDS());
break;
}
"""
self.run_cases_test(input, input2, output)

def test_validate_uop_unused_output(self):
input = """
op(OP, ( -- unused)) {
}
"""
input2 = """
op(OP, ( -- foo)) {
foo = NULL;
}
"""
output = """
case OP: {
JitOptSymbol *foo;
foo = NULL;
stack_pointer[0] = foo;
stack_pointer += 1;
assert(WITHIN_STACK_BOUNDS());
break;
}
"""
self.run_cases_test(input, input2, output)

input = """
op(OP, ( -- foo)) {
foo = NULL;
}
"""
input2 = """
op(OP, ( -- unused)) {
}
"""
output = """
case OP: {
stack_pointer += 1;
assert(WITHIN_STACK_BOUNDS());
break;
}
"""
self.run_cases_test(input, input2, output)

def test_validate_uop_input_size_mismatch(self):
input = """
op(OP, (arg1[2] -- )) {
}
"""
input2 = """
op(OP, (arg1[4] -- )) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Inputs must have equal sizes"):
self.run_cases_test(input, input2, output)

def test_validate_uop_output_size_mismatch(self):
input = """
op(OP, ( -- out[2])) {
}
"""
input2 = """
op(OP, ( -- out[4])) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Outputs must have equal sizes"):
self.run_cases_test(input, input2, output)

def test_validate_uop_unused_size_mismatch(self):
input = """
op(OP, (foo[2] -- )) {
}
"""
input2 = """
op(OP, (unused[4] -- )) {
}
"""
output = """
"""
with self.assertRaisesRegex(SyntaxError,
"Inputs must have equal sizes"):
self.run_cases_test(input, input2, output)

if __name__ == "__main__":
unittest.main()
Loading
Loading