import numpy as np
from spglib import get_magnetic_symmetry_dataset, get_symmetry_dataset

cell = (
    np.array([
        [-2.85823589e-08,  6.41219485e+00,  4.53409959e+00],
        [ 5.55312336e+00, -3.20609746e+00,  4.53409960e+00],
        [-2.77656167e+00, -1.60304871e+00,  2.26704980e+00]
    ]),
    np.array([
        [1.36114730e-08, 1.25613580e-07, 9.99999841e-01],
        [2.49999880e-01, 2.50000091e-01, 1.71296187e-07],
        [2.50000020e-01, 8.12272043e-08, 4.99999899e-01],
        [7.45180680e-08, 2.50000089e-01, 4.99999615e-01],
        [9.99999885e-01, 4.99999961e-01, 7.51512471e-08],
        [2.49999937e-01, 7.49999899e-01, 3.42200749e-07],
        [2.49999990e-01, 4.99999997e-01, 4.99999929e-01],
        [9.99999967e-01, 7.49999890e-01, 5.00000321e-01],
        [4.99999988e-01, 9.99999973e-01, 1.07227458e-07],
        [7.49999930e-01, 2.50000088e-01, 9.99999941e-01],
        [7.49999928e-01, 9.99999913e-01, 5.00000252e-01],
        [5.00000016e-01, 2.49999991e-01, 4.99999988e-01],
        [5.00000131e-01, 4.99999971e-01, 9.99999904e-01],
        [7.50000138e-01, 7.49999926e-01, 9.99999880e-01],
        [7.50000037e-01, 5.00000064e-01, 4.99999880e-01],
        [5.00000061e-01, 7.49999939e-01, 4.99999858e-01]
    ]),
    np.array([1, 4, 4, 4, 1, 4, 4, 4, 1, 4, 4, 4, 1, 4, 4, 4]),
    np.array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])
)

dataset = get_magnetic_symmetry_dataset(cell)
print(dataset)