

"""simple docstring"""





import os
import sys
import unittest


_a							=							os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
sys.path.append(os.path.join(git_repo_path, 'utils'))

import check_dummies  # noqa: E402
from check_dummies import create_dummy_files, create_dummy_object, find_backend, read_init  # noqa: E402


# Align TRANSFORMERS_PATH in check_dummies with the current path
_a							=							os.path.join(git_repo_path, 'src', 'transformers')

_a							=							'\n{0} = None\n'

_a							=							'\nclass {0}(metaclass=DummyObject):\n    _backends = {1}\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, {1})\n'


_a							=							'\ndef {0}(*args, **kwargs):\n    requires_backends({0}, {1})\n'




class  _lowerCAmelCase  (     unittest.TestCase       ):







 """simple docstring"""







 def  _lowercase			(      self							:      Any     ):
    __lowercase								=						find_backend("    _import_structure[\"models.albert\"].append(\"AlbertTokenizerFast\")"     )
    self.assertIsNone(lowercase_     )

    __lowercase								=						find_backend("    if not is_tokenizers_available():"     )
    self.assertEqual(lowercase_, "tokenizers"     )

    __lowercase								=						find_backend("    if not is_tensorflow_text_available():"     )
    self.assertEqual(lowercase_, "tensorflow_text"     )

    __lowercase								=						find_backend("    if not (is_sentencepiece_available() and is_tokenizers_available()):"     )
    self.assertEqual(lowercase_, "sentencepiece_and_tokenizers"     )

    __lowercase								=						find_backend(
        "    if not (is_sentencepiece_available() and is_tensorflow_text_available()):"     )
    self.assertEqual(lowercase_, "sentencepiece_and_tensorflow_text"     )

    __lowercase								=						find_backend(
        "    if not (is_sentencepiece_available() and is_tokenizers_available() and is_vision_available()):"     )
    self.assertEqual(lowercase_, "sentencepiece_and_tokenizers_and_vision"     )


 def  _lowercase			(      self							:      Tuple     ):
    __lowercase								=						read_init()
    # We don't assert on the exact list of keys to allow for smooth grow of backend-specific objects
    self.assertIn("torch", lowercase_     )
    self.assertIn("tensorflow_text", lowercase_     )
    self.assertIn("sentencepiece_and_tokenizers", lowercase_     )

    # Likewise, we can't assert on the exact content of a key
    self.assertIn("BertModel", objects["torch"]     )
    self.assertIn("TFBertModel", objects["tf"]     )
    self.assertIn("FlaxBertModel", objects["flax"]     )
    self.assertIn("BertModel", objects["torch"]     )
    self.assertIn("TFBertTokenizer", objects["tensorflow_text"]     )
    self.assertIn("convert_slow_tokenizer", objects["sentencepiece_and_tokenizers"]     )


 def  _lowercase			(      self							:      int     ):
    __lowercase								=						create_dummy_object("CONSTANT", "'torch'"     )
    self.assertEqual(lowercase_, "\nCONSTANT = None\n"     )

    __lowercase								=						create_dummy_object("function", "'torch'"     )
    self.assertEqual(
        lowercase_, "\ndef function(*args, **kwargs):\n    requires_backends(function, 'torch')\n"     )

    __lowercase								=						"\nclass FakeClass(metaclass=DummyObject):\n    _backends = 'torch'\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, 'torch')\n"
    __lowercase								=						create_dummy_object("FakeClass", "'torch'"     )
    self.assertEqual(lowercase_, lowercase_     )



 def  _lowercase			(      self							:      Any     ):
    __lowercase								=						"# This file is autogenerated by the command `make fix-copies`, do not edit.\nfrom ..utils import DummyObject, requires_backends\n\n\nCONSTANT = None\n\n\ndef function(*args, **kwargs):\n    requires_backends(function, [\"torch\"])\n\n\nclass FakeClass(metaclass=DummyObject):\n    _backends = [\"torch\"]\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, [\"torch\"])\n"
    __lowercase								=						create_dummy_files({"torch": ["CONSTANT", "function", "FakeClass"]}     )
    self.assertEqual(dummy_files["torch"], lowercase_     )


"""simple docstring"""





import os
import random
import sys

from . import cryptomath_module as cryptoMath  # noqa: N812
from . import rabin_miller as rabinMiller  # noqa: N812




def   _A  (  )     ->						None:
   '''simple docstring'''


   print("Making key files...")
   make_key_files("rsa",      1024)
   print("Key files generation successful.")




def   _A  (  UpperCamelCase_					:	int)     ->						tuple[tuple[int, int], tuple[int, int]]:
   '''simple docstring'''


   print("Generating prime p...")
   __lowercase								=						rabinMiller.generate_large_prime(UpperCamelCase_)
   print("Generating prime q...")
   __lowercase								=						rabinMiller.generate_large_prime(UpperCamelCase_)
   __lowercase								=						p * q

   print("Generating e that is relatively prime to (p - 1) * (q - 1)...")
   while True:
      __lowercase								=						random.randrange(2 ** (key_size - 1),      2 ** (key_size))
      if cryptoMath.gcd(UpperCamelCase_,      (p - 1) * (q - 1)) == 1:
         break

   print("Calculating d that is mod inverse of e...")
   __lowercase								=						cryptoMath.find_mod_inverse(UpperCamelCase_,      (p - 1) * (q - 1))

   __lowercase								=						(n, e)
   __lowercase								=						(n, d)
   return (public_key, private_key)




def   _A  (  UpperCamelCase_					:	str,      UpperCamelCase_					:	int)     ->						None:
   '''simple docstring'''


   if os.path.exists(F"""{name}_pubkey.txt""") or os.path.exists(F"""{name}_privkey.txt"""):
      print("\nWARNING:")
      print(
          F"""\"{name}_pubkey.txt\" or \"{name}_privkey.txt\" already exists. \n"""
          "Use a different name or delete these files and re-run this program.")
      sys.exit()

   __lowercase		,__lowercase								=						generate_key(UpperCamelCase_)
   print(F"""\nWriting public key to file {name}_pubkey.txt...""")
   with open(F"""{name}_pubkey.txt""",      "w") as out_file:
      out_file.write(F"""{key_size},{public_key[0]},{public_key[1]}""")

   print(F"""Writing private key to file {name}_privkey.txt...""")
   with open(F"""{name}_privkey.txt""",      "w") as out_file:
      out_file.write(F"""{key_size},{private_key[0]},{private_key[1]}""")


if __name__ == "__main__":
   main()




