def test_ordinary(self): for Model in self.cov_models: model = Model(dim=1, var=0.5, len_scale=2, anis=[0.1, 1], angles=[0.5, 0, 0]) srf = SRF(model, seed=19970221) srf.set_condition(self.cond_pos[0], self.cond_val, "ordinary") field_1 = srf.unstructured(self.pos[0]) field_2 = srf.structured(self.pos[0]) for i, val in enumerate(self.cond_val): self.assertAlmostEqual(val, field_1[i], places=2) self.assertAlmostEqual(val, field_2[(i, )], places=2) for dim in self.dims[1:]: model = Model( dim=dim, var=0.5, len_scale=2, anis=[0.1, 1], angles=[0.5, 0, 0], ) srf = SRF(model, seed=19970221) srf.set_condition(self.cond_pos[:dim], self.cond_val, "ordinary") field_1 = srf.unstructured(self.pos[:dim]) field_2 = srf.structured(self.pos[:dim]) for i, val in enumerate(self.cond_val): self.assertAlmostEqual(val, field_1[i], places=2) self.assertAlmostEqual(val, field_2[dim * (i, )], places=2)
def test_calls(self): srf = SRF(self.cov_model, mean=self.mean, mode_no=self.mode_no) field = srf((self.x_tuple, self.y_tuple), seed=self.seed) field2 = srf.unstructured((self.x_tuple, self.y_tuple), seed=self.seed) self.assertAlmostEqual(field[0], srf.field[0]) self.assertAlmostEqual(field[0], field2[0]) field = srf( (self.x_tuple, self.y_tuple), seed=self.seed, mesh_type="structured", ) field2 = srf.structured((self.x_tuple, self.y_tuple), seed=self.seed) self.assertAlmostEqual(field[0, 0], srf.field[0, 0]) self.assertAlmostEqual(field[0, 0], field2[0, 0])