import numpy as np

from .common import Benchmark


class Records(Benchmark):
    def setup(self):
        self.l50 = np.arange(1000)
        self.fields_number = 10000
        self.arrays = [self.l50 for _ in range(self.fields_number)]
        self.formats = [self.l50.dtype.str for _ in range(self.fields_number)]
        self.formats_str = ','.join(self.formats)
        self.dtype_ = np.dtype(
            [
                (f'field_{i}', self.l50.dtype.str)
                for i in range(self.fields_number)
            ]
        )
        self.buffer = self.l50.tobytes() * self.fields_number

    def time_fromarrays_w_dtype(self):
        np._core.records.fromarrays(self.arrays, dtype=self.dtype_)

    def time_fromarrays_wo_dtype(self):
        np._core.records.fromarrays(self.arrays)

    def time_fromarrays_formats_as_list(self):
        np._core.records.fromarrays(self.arrays, formats=self.formats)

    def time_fromarrays_formats_as_string(self):
        np._core.records.fromarrays(self.arrays, formats=self.formats_str)

    def time_frombytes_w_dtype(self):
        np._core.records.fromstring(self.buffer, dtype=self.dtype_)

    def time_frombytes_formats_as_list(self):
        np._core.records.fromstring(self.buffer, formats=self.formats)

    def time_frombytes_formats_as_string(self):
        np._core.records.fromstring(self.buffer, formats=self.formats_str)