def test_bz_parametrize_integer(self, setup): # parametrize for single integers def func(parent, N, i): return [i / N, 0, 0] bz = BrillouinZone.parametrize(setup.s1, func, 10) assert len(bz) == 10 assert np.allclose(bz.k[-1], [9 / 10, 0, 0])
def test_bz_parametrize_list(self, setup): # parametrize for single integers def func(parent, N, i): return [i[0] / N[0], i[1] / N[1], 0] bz = BrillouinZone.parametrize(setup.s1, func, [10, 2]) assert len(bz) == 20 assert np.allclose(bz.k[-1], [9 / 10, 1 / 2, 0]) assert np.allclose(bz.k[-2], [9 / 10, 0 / 2, 0])