





import unittest

import numpy as np
import torch

from .utils_summarization import build_mask, compute_token_type_ids, process_story, truncate_or_pad



class        _lowerCAmelCase    (			unittest.TestCase							):


  '''simple docstring'''







  def UpperCamelCase_ (      self      :					Dict							):


    '''simple docstring'''





    _snake_case   :				Any														=	10







  def UpperCamelCase_ (      self      :					Dict							):


    '''simple docstring'''





    _snake_case   :				List[Any]														=	[1, 2, 3, 4]
    _snake_case   :				Optional[int]														=	[1, 2, 3, 4, 0, 0, 0, 0, 0, 0]
    self.assertEqual(truncate_or_pad(UpperCamelCase      ,				self.block_size      ,				0							)      ,				UpperCamelCase							)







  def UpperCamelCase_ (      self      :					Optional[int]							):


    '''simple docstring'''





    _snake_case   :				Optional[Any]														=	[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    _snake_case   :				int														=	[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    self.assertEqual(truncate_or_pad(UpperCamelCase      ,				self.block_size      ,				0							)      ,				UpperCamelCase							)







  def UpperCamelCase_ (      self      :					Tuple							):


    '''simple docstring'''





    _snake_case   :				Optional[Any]														=	[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
    _snake_case   :				List[Any]														=	[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    self.assertEqual(truncate_or_pad(UpperCamelCase      ,				self.block_size      ,				0							)      ,				UpperCamelCase							)







  def UpperCamelCase_ (      self      :					Any							):


    '''simple docstring'''





    _snake_case   :				Tuple														=	'It was the year of Our Lord one thousand seven hundred and\n        seventy-five.\n\nSpiritual revelations were conceded to England at that\n        favoured period, as at this.'
    _snake_case							,					_snake_case   :				List[str]														=	process_story(UpperCamelCase							)
    self.assertEqual(UpperCamelCase      ,				[]							)







  def UpperCamelCase_ (      self      :					Dict							):


    '''simple docstring'''





    _snake_case   :				Optional[int]														=	''
    _snake_case							,					_snake_case   :				Union[str, Any]														=	process_story(UpperCamelCase							)
    self.assertEqual(UpperCamelCase      ,				[]							)
    self.assertEqual(UpperCamelCase      ,				[]							)







  def UpperCamelCase_ (      self      :					Optional[Any]							):


    '''simple docstring'''





    _snake_case   :				Any														=	(
        'It was the year of Our Lord one thousand seven hundred and '
        'seventy-five\n\nSpiritual revelations were conceded to England '
        'at that favoured period, as at this.\n@highlight\n\nIt was the best of times'
    )
    _snake_case							,					_snake_case   :				Union[str, Any]														=	process_story(UpperCamelCase							)

    _snake_case   :				Union[str, Any]														=	[
        'It was the year of Our Lord one thousand seven hundred and seventy-five.',
        'Spiritual revelations were conceded to England at that favoured period, as at this.',
    ]
    self.assertEqual(UpperCamelCase      ,				UpperCamelCase							)

    _snake_case   :				int														=	['It was the best of times.']
    self.assertEqual(UpperCamelCase      ,				UpperCamelCase							)







  def UpperCamelCase_ (      self      :					int							):


    '''simple docstring'''





    _snake_case   :				Tuple														=	torch.tensor([1, 2, 3, 4]							)
    _snake_case   :				Union[str, Any]														=	torch.tensor([1, 1, 1, 1]							)
    np.testing.assert_array_equal(build_mask(UpperCamelCase      ,				0							).numpy()      ,				expected.numpy()							)







  def UpperCamelCase_ (      self      :					int							):


    '''simple docstring'''





    _snake_case   :				Optional[Any]														=	torch.tensor([1, 2, 3, 4, 23, 23, 23]							)
    _snake_case   :				List[Any]														=	torch.tensor([1, 1, 1, 1, 0, 0, 0]							)
    np.testing.assert_array_equal(build_mask(UpperCamelCase      ,				23							).numpy()      ,				expected.numpy()							)







  def UpperCamelCase_ (      self      :					Any							):


    '''simple docstring'''





    _snake_case   :				Optional[Any]														=	torch.tensor([8, 2, 3, 4, 1, 1, 1]							)
    _snake_case   :				str														=	torch.tensor([1, 1, 1, 1, 0, 0, 0]							)
    np.testing.assert_array_equal(build_mask(UpperCamelCase      ,				1							).numpy()      ,				expected.numpy()							)







  def UpperCamelCase_ (      self      :					Any							):


    '''simple docstring'''





    _snake_case   :				Dict														=	1_01
    _snake_case   :				Optional[int]														=	torch.tensor([[1, 2, 3, 4, 5, 6], [1, 2, 3, 1_01, 5, 6], [1, 1_01, 3, 4, 1_01, 6]]							)
    _snake_case   :				str														=	torch.tensor([[1, 1, 1, 1, 1, 1], [1, 1, 1, 0, 0, 0], [1, 0, 0, 0, 1, 1]]							)

    _snake_case   :				Dict														=	compute_token_type_ids(UpperCamelCase      ,				UpperCamelCase							)
    np.testing.assert_array_equal(UpperCamelCase      ,				UpperCamelCase							)











from functools import reduce

lowerCAmelCase_						   =    (
    """73167176531330624919225119674426574742355349194934"""
    """96983520312774506326239578318016984801869478851843"""
    """85861560789112949495459501737958331952853208805511"""
    """12540698747158523863050715693290963295227443043557"""
    """66896648950445244523161731856403098711121722383113"""
    """62229893423380308135336276614282806444486645238749"""
    """30358907296290491560440772390713810515859307960866"""
    """70172427121883998797908792274921901699720888093776"""
    """65727333001053367881220235421809751254540594752243"""
    """52584907711670556013604839586446706324415722155397"""
    """53697817977846174064955149290862569321978468622482"""
    """83972241375657056057490261407972968652414535100474"""
    """82166370484403199890008895243450658541227588666881"""
    """16427171479924442928230863465674813919123162824586"""
    """17866458359124566529476545682848912883142607690042"""
    """24219022671055626321111109370544217506941658960408"""
    """07198403850962455444362981230987879927244284909188"""
    """84580156166097919133875499200524063689912560717606"""
    """05886116467109405077541002256983155200055935729725"""
    """71636269561882670428252483600823257530420752963450"""
)


def     lowerCamelCase_	(      lowerCAmelCase:	str = N				)->		int:
  return max(
      # mypy cannot properly interpret reduce
      int(reduce(lambda lowerCAmelCase    ,							lowerCAmelCase				: str(int(lowerCAmelCase				) * int(lowerCAmelCase				)				)    ,							n[i : i + 13]				)				)
      for i in range(len(lowerCAmelCase				) - 12				)				)


if __name__ == "__main__":
      print(F"""{solution() = }""")




