TEST: atomic_* functions

[kennykos]

Previously, only atomic_fetch_* functions where tested. Now we test atomic_* as well.

[kennykos]

@Wilson1211 can you confirm that this change works for you're workunits? If not, please share a workunit that breaks and I will add this to the test suite.

[Wilson1211]

@kennykos
This will affect atomic_compare_exchange

Error in /shared/graphyk/pk-llp/src/bc_teampolicy.py:109: atomic_op functions take exactly 3 arguments
DEBUG AST:
Call(
func=Attribute(
value=Name(id='pk', ctx=Load()),
attr='atomic_compare_exchange',
ctx=Load()),
args=[
Name(id='labels', ctx=Load()),
List(
elts=[
Name(id='neighborIdx', ctx=Load())],
ctx=Load()),
UnaryOp(
op=USub(),
operand=Constant(value=1)),
Name(id='new_label', ctx=Load())])
old_label : int = pk.atomic_compare_exchange(labels, [neighborIdx], -1, new_label)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Translation of <ast.FunctionDef object at 0x7af6e815dd50> forward failed

[kennykos]

atomic_compare_exchange

Great, I will add this to the test suite.

[kennykos]

@Wilson1211 I added a test for atomic_compare_exchange, let me know if everything works now.

[Wilson1211]

@kennykos
All good! Works now.

[JBludau]

looks good