def test_llvm_2_function_and_mappings():
llvm_program = LLVMBuilder(initial_values, variable_names, STATES, DERIVATIVES)
llvm_program.add_external_function(eval_llvm, eval_llvm_signature, number_of_args=4, target_ids=[2, 3])
llvm_program.add_call(eval_llvm.__qualname__, ["oscillator1.mechanics.b", "oscillator1.mechanics.y",
"oscillator1.mechanics.a", "oscillator1.mechanics.y_dot"],
target_ids=[2, 3])
diff, var_func, _ = llvm_program.generate(filename)
assert approx(diff(np.array([2.1, 2.2, 2.3]))) == np.array([7., 100., 9.])
assert approx(var_func()) == np.array([2.1, 2.2, 2.3, -100., 5., 6., 7., 100., 9.])
def test_llvm_1_function_and_mapping_unordered_vars():
llvm_program = LLVMBuilder(initial_values, variable_distributed, STATES, DERIVATIVES)
llvm_program.add_external_function(eval_llvm, eval_llvm_signature, number_of_args=4, target_ids=[2, 3])
llvm_program.add_call(eval_llvm.__qualname__,
["oscillator1.mechanics.x", "oscillator1.mechanics.y",
"oscillator1.mechanics.a", "oscillator1.mechanics.y_dot"], target_ids=[2, 3])
llvm_program.add_mapping(args=["oscillator1.mechanics.a"],
targets=["oscillator1.mechanics.x_dot"])
diff, var_func, _ = llvm_program.generate(filename)
assert approx(diff(np.array([2.1, 2.2, 2.3]))) == np.array([50., -50., 4.])
assert approx(diff(np.array([2.3, 2.2, 2.1]))) == np.array([-100, 100, 4.])
def test_llvm_loop_mix():
llvm_program = LLVMBuilder(initial_values, variable_names, STATES, DERIVATIVES)
llvm_program.add_external_function(eval_llvm_mix, eval_llvm_mix_signature, number_of_args=4, target_ids=[1, 3])
llvm_program.add_set_call(eval_llvm_mix.__qualname__, [
["oscillator1.mechanics.y", "oscillator1.mechanics.z", "oscillator1.mechanics.a", "oscillator1.mechanics.b"],
["oscillator1.mechanics.c", "oscillator1.mechanics.x_dot", "oscillator1.mechanics.y_dot",
"oscillator1.mechanics.z_dot"]],
targets_ids=[1, 3])
diff, var_func, _ = llvm_program.generate(filename)
assert approx(diff(np.array([2.6, 2.2, 2.3]))) == np.array([50, 8, -50])
assert approx(var_func()) == np.array([2.6, 2.2, 2.3, 4, -50., 6., 50, 8., -50.])
def test_llvm_loop_seq():
llvm_program = LLVMBuilder(initial_values, variable_names, STATES, DERIVATIVES)
llvm_program.add_external_function(eval_llvm, eval_llvm_signature, number_of_args=4, target_ids=[2, 3])
llvm_program.add_set_call(eval_llvm.__qualname__, [
["oscillator1.mechanics.y", "oscillator1.mechanics.z", "oscillator1.mechanics.a", "oscillator1.mechanics.b"],
["oscillator1.mechanics.c", "oscillator1.mechanics.x_dot", "oscillator1.mechanics.y_dot",
"oscillator1.mechanics.z_dot"]],
targets_ids=[2, 3])
diff, var_func, _ = llvm_program.generate(filename)
assert approx(diff(np.array([2.6, 2.2, 2.3]))) == np.array([7, 50., -50])
##Note that state is not changed. States can only be changed by the solver
assert approx(var_func()) == np.array([2.6, 2.2, 2.3, 50, -50., 6., 7., 50., -50.])
def test_llvm_unordered_vars():
llvm_program = LLVMBuilder(initial_values, variable_distributed, STATES, DERIVATIVES)
llvm_program.add_external_function(eval_llvm, eval_llvm_signature, number_of_args=4, target_ids=[2, 3])
diff, var_func, _ = llvm_program.generate(filename)
assert approx(diff(np.array([2.1, 2.2, 2.3]))) == np.array([2., 3., 4.])
assert approx(var_func()) == np.array([1., 2., 3., 4., 2.1, 6., 7., 2.2, 2.3])