'''simple docstring'''

from unittest import TestCase

from datasets import Sequence, Value
from datasets.arrow_dataset import Dataset







class 				_lowercase     (				UpperCAmelCase__			):


 '''simple docstring'''






 def a					(      self   :	int							)  ->    Optional[Any]:
  return [
      {"col_1": 3, "col_2": "a"},
      {"col_1": 2, "col_2": "b"},
      {"col_1": 1, "col_2": "c"},
      {"col_1": 0, "col_2": "d"},
  ]
 def a					(      self   :	List[Any]							)  ->    Any:
  __lowerCAmelCase											=						{"""col_1""": [3, 2, 1, 0], """col_2""": ["""a""", """b""", """c""", """d"""]}
  return Dataset.from_dict(SCREAMING_SNAKE_CASE__							)
 def a					(      self   :	List[Any]							)  ->    Tuple:
  __lowerCAmelCase											=						self._create_example_records()
  __lowerCAmelCase											=						Dataset.from_list(SCREAMING_SNAKE_CASE__							)
  self.assertListEqual(dset.column_names      ,       ["""col_1""", """col_2"""]							)
  for i, r in enumerate(SCREAMING_SNAKE_CASE__							):
   self.assertDictEqual(SCREAMING_SNAKE_CASE__      ,       example_records[i]							)
 def a					(      self   :	Tuple							)  ->    List[str]:
  __lowerCAmelCase											=						self._create_example_records()
  __lowerCAmelCase											=						Dataset.from_list(SCREAMING_SNAKE_CASE__							)
  __lowerCAmelCase											=						Dataset.from_dict({k: [r[k] for r in example_records] for k in example_records[0]}							)
  self.assertEqual(dset.info      ,       dset_from_dict.info							)
 def a					(      self   :	List[str]							)  ->    List[str]:  # checks what happens with missing columns
  __lowerCAmelCase											=						[{"""col_1""": 1}, {"""col_2""": """x"""}]
  __lowerCAmelCase											=						Dataset.from_list(SCREAMING_SNAKE_CASE__							)
  self.assertDictEqual(dset[0]      ,       {"""col_1""": 1}							)
  self.assertDictEqual(dset[1]      ,       {"""col_1""": None}							)  # NB: first record is used for columns
 def a					(      self   :	Dict							)  ->    Optional[int]:  # checks if the type can be inferred from the second record
  __lowerCAmelCase											=						[{"""col_1""": []}, {"""col_1""": [1, 2]}]
  __lowerCAmelCase											=						Dataset.from_list(SCREAMING_SNAKE_CASE__							)
  self.assertEqual(dset.info.features["""col_1"""]      ,       Sequence(Value("""int64"""							)							)							)





 def a					(      self   :	Optional[Any]							)  ->    Tuple:
  __lowerCAmelCase											=						Dataset.from_list([]							)
  self.assertEqual(len(SCREAMING_SNAKE_CASE__							)      ,       0							)
  self.assertListEqual(dset.column_names      ,       []							)







'''simple docstring'''

import unittest

import numpy as np
import torch

from diffusers import PNDMPipeline, PNDMScheduler, UNetaDModel
from diffusers.utils.testing_utils import enable_full_determinism, require_torch, slow, torch_device


enable_full_determinism()







class 				_lowercase     (				unittest.TestCase			):


 '''simple docstring'''






 @property
 def a					(      self   :	Any							)  ->    int:
  torch.manual_seed(0							)
  __lowerCAmelCase											=						UNetaDModel(
      block_out_channels=(32, 64)      ,       layers_per_block=2      ,       sample_size=32      ,       in_channels=3      ,       out_channels=3      ,       down_block_types=("""DownBlock2D""", """AttnDownBlock2D""")      ,       up_block_types=("""AttnUpBlock2D""", """UpBlock2D""")      ,       )
  return model





 def a					(      self   :	Dict							)  ->    int:
  __lowerCAmelCase											=						self.dummy_uncond_unet
  __lowerCAmelCase											=						PNDMScheduler()

  __lowerCAmelCase											=						PNDMPipeline(unet=SCREAMING_SNAKE_CASE__      ,       scheduler=SCREAMING_SNAKE_CASE__							)
  pndm.to(SCREAMING_SNAKE_CASE__							)
  pndm.set_progress_bar_config(disable=SCREAMING_SNAKE_CASE__							)

  __lowerCAmelCase											=						torch.manual_seed(0							)
  __lowerCAmelCase											=						pndm(generator=SCREAMING_SNAKE_CASE__      ,       num_inference_steps=20      ,       output_type="""numpy"""							).images

  __lowerCAmelCase											=						torch.manual_seed(0							)
  __lowerCAmelCase											=						pndm(generator=SCREAMING_SNAKE_CASE__      ,       num_inference_steps=20      ,       output_type="""numpy"""      ,       return_dict=SCREAMING_SNAKE_CASE__							)[0]

  __lowerCAmelCase											=						image[0, -3:, -3:, -1]
  __lowerCAmelCase											=						image_from_tuple[0, -3:, -3:, -1]

  assert image.shape == (1, 32, 32, 3)
  __lowerCAmelCase											=						np.array([1.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 0.0, 0.0]							)

  assert np.abs(image_slice.flatten() - expected_slice							).max() < 1e-2
  assert np.abs(image_from_tuple_slice.flatten() - expected_slice							).max() < 1e-2







@slow
@require_torch
class 				_lowercase     (				unittest.TestCase			):


 '''simple docstring'''






 def a					(      self   :	Any							)  ->    Any:
  __lowerCAmelCase											=						"""google/ddpm-cifar10-32"""

  __lowerCAmelCase											=						UNetaDModel.from_pretrained(SCREAMING_SNAKE_CASE__							)
  __lowerCAmelCase											=						PNDMScheduler()

  __lowerCAmelCase											=						PNDMPipeline(unet=SCREAMING_SNAKE_CASE__      ,       scheduler=SCREAMING_SNAKE_CASE__							)
  pndm.to(SCREAMING_SNAKE_CASE__							)
  pndm.set_progress_bar_config(disable=SCREAMING_SNAKE_CASE__							)
  __lowerCAmelCase											=						torch.manual_seed(0							)
  __lowerCAmelCase											=						pndm(generator=SCREAMING_SNAKE_CASE__      ,       output_type="""numpy"""							).images

  __lowerCAmelCase											=						image[0, -3:, -3:, -1]

  assert image.shape == (1, 32, 32, 3)
  __lowerCAmelCase											=						np.array([0.1_5_6_4, 0.1_4_6_4_5, 0.1_4_0_6, 0.1_4_7_1_5, 0.1_2_4_2_5, 0.1_4_0_4_5, 0.1_3_1_1_5, 0.1_2_1_7_5, 0.1_2_5]							)

  assert np.abs(image_slice.flatten() - expected_slice							).max() < 1e-2






