Need a test for raising an error if mode is neither `uniform` or `normal`?