


'''simple docstring'''

import numpy as np
from PIL import Image





def __snake_case      (   UpperCAmelCase_		:		np.ndarray      ,       UpperCAmelCase_		:		int      ,       UpperCAmelCase_		:		int	):
 lowerCamelCase_  				=						np.array(UpperCAmelCase_	)
 if arr.shape[0] != arr.shape[1]:
  raise ValueError("The input array is not a square matrix"	)
 lowerCamelCase_  				=						0
 lowerCamelCase_  				=						0
 lowerCamelCase_  				=						0
 lowerCamelCase_  				=						0

 # compute the shape of the output matrix
 lowerCamelCase_  				=						(arr.shape[0] - size) // stride + 1
 # initialize the output matrix with zeros of shape maxpool_shape
 lowerCamelCase_  				=						np.zeros((maxpool_shape, maxpool_shape)	)

 while i < arr.shape[0]:
  if i + size > arr.shape[0]:
   # if the end of the matrix is reached, break
   break
  while j < arr.shape[1]:
   # if the end of the matrix is reached, break
   if j + size > arr.shape[1]:
    break
   # compute the maximum of the pooling matrix
   lowerCamelCase_  				=						np.max(arr[i : i + size, j : j + size]	)
   # shift the pooling matrix by stride of column pixels
   j += stride
   mat_j += 1

  # shift the pooling matrix by stride of row pixels
  i += stride
  mat_i += 1

  # reset the column index to 0
  lowerCamelCase_  				=						0
  lowerCamelCase_  				=						0

 return updated_arr





def __snake_case      (   UpperCAmelCase_		:		np.ndarray      ,       UpperCAmelCase_		:		int      ,       UpperCAmelCase_		:		int	):
 lowerCamelCase_  				=						np.array(UpperCAmelCase_	)
 if arr.shape[0] != arr.shape[1]:
  raise ValueError("The input array is not a square matrix"	)
 lowerCamelCase_  				=						0
 lowerCamelCase_  				=						0
 lowerCamelCase_  				=						0
 lowerCamelCase_  				=						0

 # compute the shape of the output matrix
 lowerCamelCase_  				=						(arr.shape[0] - size) // stride + 1
 # initialize the output matrix with zeros of shape avgpool_shape
 lowerCamelCase_  				=						np.zeros((avgpool_shape, avgpool_shape)	)

 while i < arr.shape[0]:
  # if the end of the matrix is reached, break
  if i + size > arr.shape[0]:
   break
  while j < arr.shape[1]:
   # if the end of the matrix is reached, break
   if j + size > arr.shape[1]:
    break
   # compute the average of the pooling matrix
   lowerCamelCase_  				=						int(np.average(arr[i : i + size, j : j + size]	)	)
   # shift the pooling matrix by stride of column pixels
   j += stride
   mat_j += 1

  # shift the pooling matrix by stride of row pixels
  i += stride
  mat_i += 1
  # reset the column index to 0
  lowerCamelCase_  				=						0
  lowerCamelCase_  				=						0

 return updated_arr


# Main Function
if __name__ == "__main__":
     from doctest import testmod

     testmod(name="""avgpooling""", verbose=True)

     # Loading the image
     a_						:       Any      			=				Image.open("""path_to_image""")

     # Converting the image to numpy array and maxpooling, displaying the result
     # Ensure that the image is a square matrix

     Image.fromarray(maxpooling(np.array(image), size=3, stride=2)).show()

     # Converting the image to numpy array and averagepooling, displaying the result
     # Ensure that the image is a square matrix

     Image.fromarray(avgpooling(np.array(image), size=3, stride=2)).show()







'''simple docstring'''

import contextlib
import os
import sqlitea

import pytest

from datasets import Dataset, Features, Value
from datasets.io.sql import SqlDatasetReader, SqlDatasetWriter

from ..utils import assert_arrow_memory_doesnt_increase, assert_arrow_memory_increases, require_sqlalchemy





def __snake_case      (   UpperCAmelCase_		:		Optional[int]      ,       UpperCAmelCase_		:		Dict	):
 assert isinstance(UpperCAmelCase_      ,       UpperCAmelCase_	)
 assert dataset.num_rows == 4
 assert dataset.num_columns == 3
 assert dataset.column_names == ["col_1", "col_2", "col_3"]
 for feature, expected_dtype in expected_features.items():
  assert dataset.features[feature].dtype == expected_dtype





@require_sqlalchemy
@pytest.mark.parametrize("keep_in_memory"      ,       [False, True]	)
def __snake_case      (   UpperCAmelCase_		:		Any      ,       UpperCAmelCase_		:		Any      ,       UpperCAmelCase_		:		List[str]      ,       UpperCAmelCase_		:		Union[str, Any]	):
 lowerCamelCase_  				=						tmp_path / "cache"
 lowerCamelCase_  				=						{"col_1": "string", "col_2": "int64", "col_3": "float64"}
 with assert_arrow_memory_increases() if keep_in_memory else assert_arrow_memory_doesnt_increase():
  lowerCamelCase_  				=						SqlDatasetReader(
      "dataset"      ,       "sqlite:///" + sqlite_path      ,       cache_dir=UpperCAmelCase_      ,       keep_in_memory=UpperCAmelCase_	).read()
 _check_sql_dataset(UpperCAmelCase_      ,       UpperCAmelCase_	)





@require_sqlalchemy
@pytest.mark.parametrize(
    "features"      ,       [
        None,
        {"col_1": "string", "col_2": "int64", "col_3": "float64"},
        {"col_1": "string", "col_2": "string", "col_3": "string"},
        {"col_1": "int32", "col_2": "int32", "col_3": "int32"},
        {"col_1": "float32", "col_2": "float32", "col_3": "float32"},
    ]      ,       )
def __snake_case      (   UpperCAmelCase_		:		Any      ,       UpperCAmelCase_		:		Optional[Any]      ,       UpperCAmelCase_		:		List[str]      ,       UpperCAmelCase_		:		str	):
 lowerCamelCase_  				=						tmp_path / "cache"
 lowerCamelCase_  				=						{"col_1": "string", "col_2": "int64", "col_3": "float64"}
 lowerCamelCase_  				=						features.copy() if features else default_expected_features
 lowerCamelCase_  				=						(
     Features({feature: Value(UpperCAmelCase_	) for feature, dtype in features.items()}	) if features is not None else None
 )
 lowerCamelCase_  				=						SqlDatasetReader("dataset"      ,       "sqlite:///" + sqlite_path      ,       features=UpperCAmelCase_      ,       cache_dir=UpperCAmelCase_	).read()
 _check_sql_dataset(UpperCAmelCase_      ,       UpperCAmelCase_	)





def __snake_case      (   UpperCAmelCase_		:		Union[str, Any]	):
 with contextlib.closing(sqlitea.connect(UpperCAmelCase_	)	) as con:
  lowerCamelCase_  				=						con.cursor()
  cur.execute("SELECT * FROM dataset"	)
  for row in cur:
   yield row





@require_sqlalchemy
def __snake_case      (   UpperCAmelCase_		:		Dict      ,       UpperCAmelCase_		:		Optional[Any]      ,       UpperCAmelCase_		:		List[str]	):
 lowerCamelCase_  				=						tmp_path / "cache"
 lowerCamelCase_  				=						os.path.join(UpperCAmelCase_      ,       "tmp.sql"	)
 lowerCamelCase_  				=						SqlDatasetReader("dataset"      ,       "sqlite:///" + sqlite_path      ,       cache_dir=UpperCAmelCase_	).read()
 SqlDatasetWriter(UpperCAmelCase_      ,       "dataset"      ,       "sqlite:///" + output_sqlite_path      ,       num_proc=1	).write()

 lowerCamelCase_  				=						iter_sql_file(UpperCAmelCase_	)
 lowerCamelCase_  				=						iter_sql_file(UpperCAmelCase_	)

 for rowa, rowa in zip(UpperCAmelCase_      ,       UpperCAmelCase_	):
  assert rowa == rowa





@require_sqlalchemy
def __snake_case      (   UpperCAmelCase_		:		Union[str, Any]      ,       UpperCAmelCase_		:		Union[str, Any]      ,       UpperCAmelCase_		:		Tuple	):
 lowerCamelCase_  				=						tmp_path / "cache"
 lowerCamelCase_  				=						os.path.join(UpperCAmelCase_      ,       "tmp.sql"	)
 lowerCamelCase_  				=						SqlDatasetReader("dataset"      ,       "sqlite:///" + sqlite_path      ,       cache_dir=UpperCAmelCase_	).read()
 SqlDatasetWriter(UpperCAmelCase_      ,       "dataset"      ,       "sqlite:///" + output_sqlite_path      ,       num_proc=2	).write()

 lowerCamelCase_  				=						iter_sql_file(UpperCAmelCase_	)
 lowerCamelCase_  				=						iter_sql_file(UpperCAmelCase_	)

 for rowa, rowa in zip(UpperCAmelCase_      ,       UpperCAmelCase_	):
  assert rowa == rowa







@require_sqlalchemy
def __snake_case      (   UpperCAmelCase_		:		Any      ,       UpperCAmelCase_		:		Tuple      ,       UpperCAmelCase_		:		Tuple	):
 lowerCamelCase_  				=						tmp_path / "cache"
 lowerCamelCase_  				=						os.path.join(UpperCAmelCase_      ,       "tmp.sql"	)
 lowerCamelCase_  				=						SqlDatasetReader("dataset"      ,       "sqlite:///" + sqlite_path      ,       cache_dir=UpperCAmelCase_	).read()
 with pytest.raises(UpperCAmelCase_	):
  SqlDatasetWriter(UpperCAmelCase_      ,       "dataset"      ,       "sqlite:///" + output_sqlite_path      ,       num_proc=0	).write()



