def METHOD_NAME(self):
    atom = """
        Li    5.281899    1.082489    0.642479
        Li   18.499727    3.791392    2.250266
    """
    a = [[8.3774538040, 0.0000000000, 0.0000000000],
         [7.7020859454, 3.2953913771, 0.0000000000],
         [7.7020859454, 1.5784896835, 2.8927451749]]
    basis = 'gth-dzvp-molopt-sr'
    kmesh = [2,] * 3
    sos_ks, irrep_ids_ks = get_so(atom, a, basis, kmesh)
    assert irrep_ids_ks == [[0, 3, 4, 5],
                            [0, 1, 2, 3],
                            [0, 1, 2, 3],
                            [0, 3, 4, 5]]
    norb_ks = get_norb(sos_ks)
    assert norb_ks == [[4, 4, 2, 2],
                       [5, 1, 1, 5],
                       [5, 1, 1, 5],
                       [4, 4, 2, 2]]