def METHOD_NAME(self):
    atom = """
        Si  0.0 0.0 0.0
    """
    a = np.eye(3) * 2.
    basis = 'gth-dzvp-molopt-sr'
    kmesh = [2,] * 3
    sos_ks, irrep_ids_ks = get_so(atom, a, basis, kmesh)
    assert irrep_ids_ks == [[0, 4, 6, 9],
                            [0, 1, 2, 7, 8, 9],
                            [0, 1, 2, 7, 8, 9],
                            [0, 4, 6, 9]]
    norb_ks = get_norb(sos_ks)
    assert norb_ks == [[2, 2, 3, 6], [3, 1, 1, 2, 2, 4],
                       [3, 1, 1, 2, 2, 4], [2, 2, 3, 6]]