def test_select_x():
assert utils.select_x(None) is None
def _check(d, expected):
x = utils.select_x(d)
assert x == expected
data = dict(col1=[1.0, 2.0, 3.0], # Q
col2=['A', 'B', 'C'], # N
col3=pd.date_range('2012', periods=3, freq='A')) # T
_check(data, 'col3')
data = dict(col1=[1.0, 2.0, 3.0], # Q
col2=['A', 'B', 'C']) # N
_check(data, 'col2')
data = dict(col1=[1.0, 2.0, 3.0]) # Q
_check(data, 'col1')
# Custom order
data = dict(col1=[1.0, 2.0, 3.0], # Q
col2=['A', 'B', 'C'], # N
col3=pd.date_range('2012', periods=3, freq='A'), # T
col4=pd.date_range('2012', periods=3, freq='A')) # T
selected_x = utils.select_x(data, ['N', 'T', 'Q', 'O'])
assert selected_x == "col2"
# Len < 1
assert utils.select_x(dict()) is None
def test_select_x():
assert utils.select_x(None) is None
def _check(d, expected):
x = utils.select_x(d)
assert x == expected
data = dict(
col1=[1.0, 2.0, 3.0], # Q
col2=['A', 'B', 'C'], # N
col3=pd.date_range('2012', periods=3, freq='A')) # T
_check(data, 'col3')
data = dict(
col1=[1.0, 2.0, 3.0], # Q
col2=['A', 'B', 'C']) # N
_check(data, 'col2')
data = dict(col1=[1.0, 2.0, 3.0]) # Q
_check(data, 'col1')
# Custom order
data = dict(
col1=[1.0, 2.0, 3.0], # Q
col2=['A', 'B', 'C'], # N
col3=pd.date_range('2012', periods=3, freq='A'), # T
col4=pd.date_range('2012', periods=3, freq='A')) # T
selected_x = utils.select_x(data, ['N', 'T', 'Q', 'O'])
assert selected_x == "col2"
# Len < 1
assert utils.select_x(dict()) is None
def _check(d, expected):
x = utils.select_x(d)
assert x == expected
def _check(d, expected):
x = utils.select_x(d)
assert x == expected