code: def test_export(self, run_eagerly, export_dtype, export_expansion):
      x = np.array([[12345, 34342]])
  
      context = tf_execution_context(run_eagerly)
      with context.scope():
          ek, dk = paillier.gen_keypair()
          assert isinstance(ek, paillier.EncryptionKey)
          assert isinstance(dk, paillier.DecryptionKey)
          n_exported = ek.export(export_dtype)
          assert isinstance(n_exported, tf.Tensor)
          assert n_exported.dtype == export_dtype
          assert n_exported.shape == (1, 1), n_exported.shape
          p_exported, q_exported = dk.export(export_dtype)
          assert isinstance(p_exported, tf.Tensor)
          assert p_exported.dtype == export_dtype
          assert p_exported.shape == (1, 1), p_exported.shape
          assert isinstance(q_exported, tf.Tensor)
          assert q_exported.dtype == export_dtype
          assert q_exported.shape == (1, 1), q_exported.shape
  
          r = paillier.gen_randomness(ek, shape=x.shape)
          assert isinstance(r, paillier.Randomness)
          r_exported = r.export(export_dtype)
          assert isinstance(r_exported, tf.Tensor)
          assert r_exported.dtype == export_dtype
          assert r_exported.shape == x.shape + export_expansion
  
          c = paillier.encrypt(ek, x, r)
          assert isinstance(c, paillier.Ciphertext)
          c_exported = c.export(export_dtype)
          assert isinstance(c_exported, tf.Tensor)
          assert c_exported.dtype == export_dtype
          assert c_exported.shape == x.shape + export_expansion
  
file_path: tf-encrypted/primitives/tf_encrypted/primitives/paillier/primitives_test.py
imports: import unittest
  import numpy
  import tensorflow
  from absl.testing import parameterized
  from tf_encrypted.primitives import paillier
  from tf_encrypted.test import tf_execution_context
label_desc: Use image node_id 1 for calling the EncryptionTest obj's underlying member method code with example usage: obj.test_export(run_eagerly, export_dtype, export_expansion) without return types