test(generator): add test cases for chebyshev nodes
@@ -106,6 +106,10 @@ def test_generator1d():
x = generator.getter()
assert _check_shape_and_grad(generator, size, x)
+ generator = Generator1D(size=size, t_min=np.log10(0.1), t_max=np.log10(2.0), method='chebyshev')
+ x = generator.getter()
+ assert _check_shape_and_grad(generator, size, x)
+
with raises(ValueError):
generator = Generator1D(size=size, t_min=0.0, t_max=2.0, method='magic')
