from vbench.benchmark import Benchmark
from datetime import datetime

common_setup = """from pandas_vb_common import *
"""

#----------------------------------------------------------------------
# Series constructors

setup = common_setup + """
data = np.random.randn(100)
index = Index(np.arange(100))
"""

ctor_series_ndarray = \
    Benchmark("Series(data, index=index)", setup=setup,
              name='series_constructor_ndarray')

setup = common_setup + """
arr = np.random.randn(100, 100)
"""

ctor_frame_ndarray = \
    Benchmark("DataFrame(arr)", setup=setup,
              name='frame_constructor_ndarray')

setup = common_setup + """
data = np.array(['foo', 'bar', 'baz'], dtype=object)
"""

ctor_index_array_string = Benchmark('Index(data)', setup=setup)

# index constructors
setup = common_setup + """
s = Series([Timestamp('20110101'),Timestamp('20120101'),Timestamp('20130101')]*1000)
"""
index_from_series_ctor = Benchmark('Index(s)', setup=setup)

dtindex_from_series_ctor = Benchmark('DatetimeIndex(s)', setup=setup)