def test_infer_shape(self, x, inp, axis): for params in self.op_params: if not params[1]: continue if params[0]: f = aet.unique(x, *params, axis=axis)[2] else: f = aet.unique(x, *params, axis=axis)[1] self._compile_and_check( [x], [f], [inp], Unique, )
def test_basic_vector(self, x, inp, axis): list_outs_expected = [ np.unique(inp, axis=axis), np.unique(inp, True, axis=axis), np.unique(inp, False, True, axis=axis), np.unique(inp, True, True, axis=axis), np.unique(inp, False, False, True, axis=axis), np.unique(inp, True, False, True, axis=axis), np.unique(inp, False, True, True, axis=axis), np.unique(inp, True, True, True, axis=axis), ] for params, outs_expected in zip(self.op_params, list_outs_expected): out = aet.unique(x, *params, axis=axis) f = aesara.function(inputs=[x], outputs=out) outs = f(inp) for out, out_exp in zip(outs, outs_expected): utt.assert_allclose(out, out_exp)