def test_llvm_1_to_1_mapping_parameter():
llvm_program = LLVMBuilder(initial_values, variable_names, STATES, DERIVATIVES)
llvm_program.add_mapping(["oscillator1.mechanics.b"], ["oscillator1.mechanics.x_dot"])
diff, var_func, _ = llvm_program.generate(filename)
assert approx(diff(np.array([2.1, 2.2, 2.3]))) == np.array([5, 8, 9.])
def test_llvm_idx_write():
llvm_program = LLVMBuilder(initial_values, variable_names, STATES, DERIVATIVES)
llvm_program.add_mapping(["oscillator1.mechanics.b"],
["oscillator1.mechanics.x_dot", "oscillator1.mechanics.y_dot"])
diff, var_func, var_write = llvm_program.generate(filename)
var_write(100, 4)
assert approx(var_func()) == np.array([1.0, 2.0, 3.0, 4., 100., 6., 7., 8., 9.])
assert approx(diff(np.array([2.6, 2.2, 2.3]))) == np.array([100, 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.])