
import tempfile
import unittest

from make_student import create_student_by_copying_alternating_layers

from transformers import AutoConfig
from transformers.file_utils import cached_property
from transformers.testing_utils import require_torch


_A   =			"sshleifer/bart-tiny-random"
_A   =			"patrickvonplaten/t5-tiny-random"







@require_torch
class 					lowerCamelCase		(  unittest.TestCase  ):





     @cached_property
     def 		UpperCAmelCase(self				:   List[str]       )						->						Tuple:
         return AutoConfig.from_pretrained(lowercase_       )





     def 		UpperCAmelCase(self				:   List[str]       )						->						Optional[int]:
         snake_case					,							*snake_case								= create_student_by_copying_alternating_layers(lowercase_    ,   tempfile.mkdtemp()    ,   e=1    ,   d=1       )
         self.assertEqual(student.config.num_hidden_layers    ,   1       )





     def 		UpperCAmelCase(self				:   int       )						->						List[Any]:
         snake_case					,							*snake_case								= create_student_by_copying_alternating_layers(lowercase_    ,   tempfile.mkdtemp()    ,   e=1    ,   d=lowercase_       )





     def 		UpperCAmelCase(self				:   int       )						->						Tuple:
         snake_case					,							*snake_case								= create_student_by_copying_alternating_layers(lowercase_    ,   tempfile.mkdtemp()    ,   e=1    ,   d=lowercase_       )
         self.assertEqual(student.config.encoder_layers    ,   1       )
         self.assertEqual(student.config.decoder_layers    ,   self.teacher_config.encoder_layers       )





     def 		UpperCAmelCase(self				:   List[str]       )						->						Union[str, Any]:
         snake_case					,							*snake_case								= create_student_by_copying_alternating_layers(lowercase_    ,   tempfile.mkdtemp()    ,   e=1    ,   d=1       )
         self.assertEqual(student.config.encoder_layers    ,   1       )
         self.assertEqual(student.config.decoder_layers    ,   1       )





     def 		UpperCAmelCase(self				:   Optional[Any]       )						->						int:

         with self.assertRaises(lowercase_       ):
             create_student_by_copying_alternating_layers(lowercase_    ,   tempfile.mkdtemp()    ,   e=lowercase_    ,   d=lowercase_       )






import unittest

from transformers.utils.backbone_utils import (
    BackboneMixin,
    get_aligned_output_features_output_indices,
    verify_out_features_out_indices,
)







class 					lowerCamelCase		(  unittest.TestCase  ):





     def 		UpperCAmelCase(self				:   Tuple       )						->						List[str]:
         snake_case								= ["a", "b", "c"]

         # Defaults to last layer if both are None
         snake_case					,							snake_case								= get_aligned_output_features_output_indices(_A    ,   _A    ,   _A       )
         self.assertEqual(_A    ,   ["c"]       )
         self.assertEqual(_A    ,   [2]       )

         # Out indices set to match out features
         snake_case					,							snake_case								= get_aligned_output_features_output_indices(["a", "c"]    ,   _A    ,   _A       )
         self.assertEqual(_A    ,   ["a", "c"]       )
         self.assertEqual(_A    ,   [0, 2]       )

         # Out features set to match out indices
         snake_case					,							snake_case								= get_aligned_output_features_output_indices(_A    ,   [0, 2]    ,   _A       )
         self.assertEqual(_A    ,   ["a", "c"]       )
         self.assertEqual(_A    ,   [0, 2]       )

         # Out features selected from negative indices
         snake_case					,							snake_case								= get_aligned_output_features_output_indices(_A    ,   [-3, -1]    ,   _A       )
         self.assertEqual(_A    ,   ["a", "c"]       )
         self.assertEqual(_A    ,   [-3, -1]       )





     def 		UpperCAmelCase(self				:   Optional[int]       )						->						str:
         # Stage names must be set
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(["a", "b"]    ,   (0, 1)    ,   _A       )

         # Out features must be a list
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(("a", "b")    ,   (0, 1)    ,   ["a", "b"]       )

         # Out features must be a subset of stage names
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(["a", "b"]    ,   (0, 1)    ,   ["a"]       )

         # Out indices must be a list or tuple
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(_A    ,   0    ,   ["a", "b"]       )

         # Out indices must be a subset of stage names
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(_A    ,   (0, 1)    ,   ["a"]       )

         # Out features and out indices must be the same length
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(["a", "b"]    ,   (0,)    ,   ["a", "b", "c"]       )

         # Out features should match out indices
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(["a", "b"]    ,   (0, 2)    ,   ["a", "b", "c"]       )

         # Out features and out indices should be in order
         with self.assertRaises(_A       ):
             verify_out_features_out_indices(["b", "a"]    ,   (0, 1)    ,   ["a", "b"]       )

         # Check passes with valid inputs
         verify_out_features_out_indices(["a", "b", "d"]    ,   (0, 1, -1)    ,   ["a", "b", "c", "d"]       )





     def 		UpperCAmelCase(self				:   List[str]       )						->						str:
         snake_case								= BackboneMixin()

         snake_case								= ["a", "b", "c"]
         snake_case								= ["a", "c"]
         snake_case								= [0, 2]

         # Check that the output features and indices are set correctly
         self.assertEqual(backbone.out_features    ,   ["a", "c"]       )
         self.assertEqual(backbone.out_indices    ,   [0, 2]       )

         # Check out features and indices are updated correctly
         snake_case								= ["a", "b"]
         self.assertEqual(backbone.out_features    ,   ["a", "b"]       )
         self.assertEqual(backbone.out_indices    ,   [0, 1]       )

         snake_case								= [-3, -1]
         self.assertEqual(backbone.out_features    ,   ["a", "c"]       )
         self.assertEqual(backbone.out_indices    ,   [-3, -1]       )

