c59e516bb0a4fd07dc86def25564c67922e15b52,tests/test_private_compare.py,TestPrivateCompare,_core_test,#TestPrivateCompare#Any#,21

Before Change

        expected = np.bitwise_xor(x > r, beta.astype(bool)).astype(np.int32)

        res = _private_compare(
            prot,
            x_bits=PondPrivateTensor(
                prot,
                *prot._share(val_dtype.tensor(tf.convert_to_tensor(x, dtype=val_dtype.native_type)).bits(bit_dtype)),
                False),
            r=PondPublicTensor(
                prot,
                val_dtype.tensor(tf.convert_to_tensor(r, dtype=val_dtype.native_type)),
                val_dtype.tensor(tf.convert_to_tensor(r, dtype=val_dtype.native_type)),
                False),
            beta=PondPublicTensor(
                prot,
                bit_dtype.tensor(tf.convert_to_tensor(beta, dtype=bit_dtype.native_type)),
                bit_dtype.tensor(tf.convert_to_tensor(beta, dtype=bit_dtype.native_type)),
                False)
        )

        with tfe.Session() as sess:
            actual = sess.run(res.reveal().value_on_0.to_native())
            np.testing.assert_array_equal(actual, expected)

After Change

    ], dtype=np.int32).reshape(2, 2, 2)

    expected = np.bitwise_xor(x > r, beta.astype(bool)).astype(np.int32)
    x_native = tf.convert_to_tensor(x, dtype=val_dtype.native_type)
    x_bits_preshare = val_dtype.tensor(x_native).bits(bit_dtype)
    x_bits = prot._share(x_bits_preshare)  // pylint: disable=protected-access

    r_native = tf.convert_to_tensor(r, dtype=val_dtype.native_type)

Instances