def test_initialize_power_batch(self):
kernel = LinearTruncatedFidelityKernel(
fidelity_dims=[1, 2], dimension=3, batch_shape=torch.Size([2])
)
power_init = torch.tensor([1, 2], dtype=torch.float)
kernel.initialize(power=power_init)
actual_value = power_init.view_as(kernel.power)
self.assertLess(torch.norm(kernel.power - actual_value), 1e-5)
def test_initialize_power(self):
kernel = LinearTruncatedFidelityKernel(fidelity_dims=[1, 2], dimension=3)
kernel.initialize(power=1)
actual_value = torch.tensor(1, dtype=torch.float).view_as(kernel.power)
self.assertLess(torch.norm(kernel.power - actual_value), 1e-5)