




'''simple docstring'''







import unittest
from dataclasses import dataclass
from typing import List, Union

import numpy as np
import PIL.Image

from diffusers.utils.outputs import BaseOutput

@dataclass
class   _snake_case( UpperCAmelCase			):
 __snake_case:       Union[List[PIL.Image.Image], np.ndarray]




class   _snake_case( unittest.TestCase			):






 def 	_UpperCamelCase   (self					:     Dict						)					->      List[str]:






    """simple docstring"""


    A__         =    CustomOutput(images=np.random.rand(1						,		3						,		4						,		4						)						)

    # check every way of getting the attribute
    assert isinstance(outputs.images						,		np.ndarray						)
    assert outputs.images.shape == (1, 3, 4, 4)
    assert isinstance(outputs['images']						,		np.ndarray						)
    assert outputs["images"].shape == (1, 3, 4, 4)
    assert isinstance(outputs[0]						,		np.ndarray						)
    assert outputs[0].shape == (1, 3, 4, 4)

    # test with a non-tensor attribute
    A__         =    CustomOutput(images=[PIL.Image.new('RGB'						,		(4, 4)						)]						)

    # check every way of getting the attribute
    assert isinstance(outputs.images						,		a						)
    assert isinstance(outputs.images[0]						,		PIL.Image.Image						)
    assert isinstance(outputs['images']						,		a						)
    assert isinstance(outputs['images'][0]						,		PIL.Image.Image						)
    assert isinstance(outputs[0]						,		a						)
    assert isinstance(outputs[0][0]						,		PIL.Image.Image						)






 def 	_UpperCamelCase   (self					:     Any						)					->      Union[str, Any]:






    """simple docstring"""


    A__         =    CustomOutput({'images': np.random.rand(1						,		3						,		4						,		4						)}						)

    # check every way of getting the attribute
    assert isinstance(outputs.images						,		np.ndarray						)
    assert outputs.images.shape == (1, 3, 4, 4)
    assert isinstance(outputs['images']						,		np.ndarray						)
    assert outputs["images"].shape == (1, 3, 4, 4)
    assert isinstance(outputs[0]						,		np.ndarray						)
    assert outputs[0].shape == (1, 3, 4, 4)

    # test with a non-tensor attribute
    A__         =    CustomOutput({'images': [PIL.Image.new('RGB'						,		(4, 4)						)]}						)

    # check every way of getting the attribute
    assert isinstance(outputs.images						,		a						)
    assert isinstance(outputs.images[0]						,		PIL.Image.Image						)
    assert isinstance(outputs['images']						,		a						)
    assert isinstance(outputs['images'][0]						,		PIL.Image.Image						)
    assert isinstance(outputs[0]						,		a						)
    assert isinstance(outputs[0][0]						,		PIL.Image.Image						)








'''simple docstring'''







import hashlib
import importlib.util
import json
import os
import pathlib
from types import ModuleType

import pytest
import requests

lowerCAmelCase_    	=							pathlib.Path.cwd().joinpath('''project_euler''')
lowerCAmelCase_    	=							pathlib.Path.cwd().joinpath(
    '''scripts''', '''project_euler_answers.json'''
)

with open(PROJECT_EULER_ANSWERS_PATH) as file_handle:
    lowerCAmelCase_    	=							json.load(file_handle)







def 	_A			(   UpperCAmelCase	):






   '''simple docstring'''







   A__         =    importlib.util.spec_from_file_location(file_path.name			,str(UpperCAmelCase	)	)
   A__         =    importlib.util.module_from_spec(UpperCAmelCase	)  # type: ignore
   spec.loader.exec_module(UpperCAmelCase	)  # type: ignore
   return module







def 	_A			(   ):






   '''simple docstring'''







   A__         =    []
   for problem_dir_path in PROJECT_EULER_DIR_PATH.iterdir():
      if problem_dir_path.is_file() or problem_dir_path.name.startswith('_'	):
         continue
      for file_path in problem_dir_path.iterdir():
         if file_path.suffix != ".py" or file_path.name.startswith(('_', 'test')	):
            continue
         solution_file_paths.append(UpperCAmelCase	)
   return solution_file_paths







def 	_A			(   ):






   '''simple docstring'''







   with open(os.environ['GITHUB_EVENT_PATH']	) as file:
      A__         =    json.load(UpperCAmelCase	)
   return event["pull_request"]["url"] + "/files"







def 	_A			(   ):






   '''simple docstring'''







   A__         =    []
   A__         =    {
       'Accept': 'application/vnd.github.v3+json',
       'Authorization': 'token ' + os.environ['GITHUB_TOKEN'],
   }
   A__         =    requests.get(get_files_url()			,headers=UpperCAmelCase	).json()
   for file in files:
      A__         =    pathlib.Path.cwd().joinpath(file['filename']	)
      if (
          filepath.suffix != ".py"
          or filepath.name.startswith(('_', 'test')	)
          or not filepath.name.startswith('sol'	)
      ):
         continue
      solution_file_paths.append(UpperCAmelCase	)
   return solution_file_paths







def 	_A			(   ):






   '''simple docstring'''







   if os.environ.get('CI'	) and os.environ.get('GITHUB_EVENT_NAME'	) == "pull_request":
      # Return only if there are any, otherwise default to all solutions
      if filepaths := added_solution_file_path():
         return filepaths
   return all_solution_file_paths()







@pytest.mark.parametrize(
    'solution_path'			,collect_solution_file_paths()			,ids=lambda UpperCAmelCase	: F"""{path.parent.name}/{path.name}"""			,)
def 	_A			(   UpperCAmelCase	):






   '''simple docstring'''







   A__         =    solution_path.parent.name[8:].zfill(3	)
   A__         =    PROBLEM_ANSWERS[problem_number]
   A__         =    convert_path_to_module(UpperCAmelCase	)
   A__         =    str(solution_module.solution()	)  # type: ignore
   A__         =    hashlib.shaaaa(answer.encode()	).hexdigest()
   assert (
       answer == expected
   ), F"""Expected solution to {problem_number} to have hash {expected}, got {answer}"""


