[TEST] Add test linear move_add_past_mul for non_linear graphs and improve previous test by checking if doing job

thanks to @quetric for pointing this out

[Code gen trafo] Restructured test and function to fit the transformation into finn transformation layout