import math from transformation import rotation import numpy as np if __name__ == "__main__": t = rotation( 45 / 180 * math.pi, np.array([0, 0, 1]), np.array([[1, 1, 0], [2, 3, 4]]) ) print(t)