import os
import tempfile
from functools import partial
from unittest import TestCase
from unittest.mock import patch

import datasets
import datasets.config

from .utils import require_beam
class   _lowerCAmelCase		(     datasets.BeamBasedBuilder				):







							def 				snake_case  (				self     :    str		):
										return datasets.DatasetInfo(
										    features=datasets.Features({'''content''': datasets.Value('''string'''		)}		)	,					supervised_keys=__snake_case	,					)







							def 				snake_case  (				self     :    int	,					__snake_case     :    Optional[Any]	,					__snake_case     :    Optional[Any]		):
										return [datasets.SplitGenerator(name=datasets.Split.TRAIN	,					gen_kwargs={'''examples''': get_test_dummy_examples()}		)]







							def 				snake_case  (				self     :    int	,					__snake_case     :    int	,					__snake_case     :    Dict		):
										import apache_beam as beam

										return pipeline | "Load Examples" >> beam.Create(__snake_case		)


class   _lowerCAmelCase		(     datasets.BeamBasedBuilder				):







							def 				snake_case  (				self     :    Union[str, Any]		):
										return datasets.DatasetInfo(
										    features=datasets.Features({'''a''': datasets.Sequence({'''b''': datasets.Value('''string'''		)}		)}		)	,					supervised_keys=__snake_case	,					)







							def 				snake_case  (				self     :    Optional[int]	,					__snake_case     :    Dict	,					__snake_case     :    Union[str, Any]		):
										return [
										    datasets.SplitGenerator(name=datasets.Split.TRAIN	,					gen_kwargs={'''examples''': get_test_nested_examples()}		)
										]







							def 				snake_case  (				self     :    int	,					__snake_case     :    Optional[Any]	,					__snake_case     :    Dict		):
										import apache_beam as beam

										return pipeline | "Load Examples" >> beam.Create(__snake_case		)





def  _lowerCamelCase					(   ):
			return [(i, {"content": content}) for i, content in enumerate(['''foo''', '''bar''', '''foobar'''])]





def  _lowerCamelCase					(   ):
			return [(i, {"a": {"b": [content]}}) for i, content in enumerate(['''foo''', '''bar''', '''foobar'''])]


class   _lowerCAmelCase		(     __SCREAMING_SNAKE_CASE				):







							@require_beam
							def 				snake_case  (				self     :    str		):
										lowerCamelCase						:Any						  =     len(get_test_dummy_examples()		)
										with tempfile.TemporaryDirectory() as tmp_cache_dir:
													lowerCamelCase						:List[str]						  =     DummyBeamDataset(cache_dir=__snake_case	,					beam_runner='''DirectRunner'''		)
													builder.download_and_prepare()
													self.assertTrue(
													    os.path.exists(
													        os.path.join(__snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					F"{builder.name}-train.arrow"		)		)		)
													self.assertDictEqual(builder.info.features	,					datasets.Features({'''content''': datasets.Value('''string'''		)}		)		)
													lowerCamelCase						:int						  =     builder.as_dataset()
													self.assertEqual(dset['''train'''].num_rows	,					__snake_case		)
													self.assertEqual(dset['''train'''].info.splits['''train'''].num_examples	,					__snake_case		)
													self.assertDictEqual(dset['''train'''][0]	,					get_test_dummy_examples()[0][1]		)
													self.assertDictEqual(
													    dset['''train'''][expected_num_examples - 1]	,					get_test_dummy_examples()[expected_num_examples - 1][1]		)
													self.assertTrue(
													    os.path.exists(os.path.join(__snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					'''dataset_info.json'''		)		)		)
													del dset







							@require_beam
							def 				snake_case  (				self     :    Optional[int]		):
										import apache_beam as beam

										lowerCamelCase						:Optional[Any]						  =     beam.io.parquetio.WriteToParquet

										lowerCamelCase						:Dict						  =     len(get_test_dummy_examples()		)
										with tempfile.TemporaryDirectory() as tmp_cache_dir:
													lowerCamelCase						:Tuple						  =     DummyBeamDataset(cache_dir=__snake_case	,					beam_runner='''DirectRunner'''		)
													with patch('''apache_beam.io.parquetio.WriteToParquet'''		) as write_parquet_mock:
																lowerCamelCase						:Optional[Any]						  =     partial(__snake_case	,					num_shards=2		)
																builder.download_and_prepare()
													self.assertTrue(
													    os.path.exists(
													        os.path.join(
													            __snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					F"{builder.name}-train-00000-of-00002.arrow"		)		)		)
													self.assertTrue(
													    os.path.exists(
													        os.path.join(
													            __snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					F"{builder.name}-train-00000-of-00002.arrow"		)		)		)
													self.assertDictEqual(builder.info.features	,					datasets.Features({'''content''': datasets.Value('''string'''		)}		)		)
													lowerCamelCase						:int						  =     builder.as_dataset()
													self.assertEqual(dset['''train'''].num_rows	,					__snake_case		)
													self.assertEqual(dset['''train'''].info.splits['''train'''].num_examples	,					__snake_case		)
													# Order is not preserved when sharding, so we just check that all the elements are there
													self.assertListEqual(sorted(dset['''train''']['''content''']		)	,					sorted(['''foo''', '''bar''', '''foobar''']		)		)
													self.assertTrue(
													    os.path.exists(os.path.join(__snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					'''dataset_info.json'''		)		)		)
													del dset







							@require_beam
							def 				snake_case  (				self     :    Any		):
										with tempfile.TemporaryDirectory() as tmp_cache_dir:
													lowerCamelCase						:Any						  =     DummyBeamDataset(cache_dir=__snake_case		)
													self.assertRaises(datasets.builder.MissingBeamOptions	,					builder.download_and_prepare		)







							@require_beam
							def 				snake_case  (				self     :    Optional[int]		):
										lowerCamelCase						:int						  =     len(get_test_nested_examples()		)
										with tempfile.TemporaryDirectory() as tmp_cache_dir:
													lowerCamelCase						:Tuple						  =     NestedBeamDataset(cache_dir=__snake_case	,					beam_runner='''DirectRunner'''		)
													builder.download_and_prepare()
													self.assertTrue(
													    os.path.exists(
													        os.path.join(__snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					F"{builder.name}-train.arrow"		)		)		)
													self.assertDictEqual(
													    builder.info.features	,					datasets.Features({'''a''': datasets.Sequence({'''b''': datasets.Value('''string'''		)}		)}		)		)
													lowerCamelCase						:Tuple						  =     builder.as_dataset()
													self.assertEqual(dset['''train'''].num_rows	,					__snake_case		)
													self.assertEqual(dset['''train'''].info.splits['''train'''].num_examples	,					__snake_case		)
													self.assertDictEqual(dset['''train'''][0]	,					get_test_nested_examples()[0][1]		)
													self.assertDictEqual(
													    dset['''train'''][expected_num_examples - 1]	,					get_test_nested_examples()[expected_num_examples - 1][1]		)
													self.assertTrue(
													    os.path.exists(os.path.join(__snake_case	,					builder.name	,					'''default'''	,					'''0.0.0'''	,					'''dataset_info.json'''		)		)		)
													del dset




import os
from math import logaa





def  _lowerCamelCase					(   a_				: str = "base_exp.txt"):
			lowerCamelCase						:float						  =     0
			lowerCamelCase						:Optional[int]						  =     0
			for i, line in enumerate(open(os.path.join(os.path.dirname(a_)						,			a_))):
						lowerCamelCase  ,    lowerCamelCase						:Optional[int]						  =     list(map(a_						,			line.split(''',''')))
						if x * logaa(a_) > largest:
									lowerCamelCase						:List[Any]						  =     x * logaa(a_)
									lowerCamelCase						:Any						  =     i + 1
			return result


if __name__ == "__main__":
			print(solution())



