Dispersion relation of a wave packet

```
def n(wj):
if dispersion == "const": #1
return 1
if dispersion == "high": # 2
return 1 + wj * 0.01
if dispersion == "low": #3
return 1.1 - wj * 0.01
```