def test_select_y():
def _check(d, expected):
x = 'col1'
y = utils.select_y(d, x)
assert y == expected
data = dict(
col1=[1.0, 2.0, 3.0], # Chosen X
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
col5=[1.0, 2.0, 3.0]) # Q
_check(data, 'col5')
data = dict(
col1=[1.0, 2.0, 3.0], # Chosen X
col2=['A', 'B', 'C'], # N
col3=pd.date_range('2012', periods=3, freq='A')) # T
_check(data, 'col2')
data = dict(
col1=[1.0, 2.0, 3.0], # Chosen X
col2=pd.date_range('2012', periods=3, freq='A')) # T
_check(data, 'col2')
# No data
assert utils.select_y(None, "something") is None
# Len < 2
assert utils.select_y(dict(col1=[1.0, 2.0, 3.0]), "something") is None
# No x
assert utils.select_y(df, None) is None
# Custom order
data = dict(
col1=[1.0, 2.0, 3.0], # Chosen X
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
col5=[1.0, 2.0, 3.0], # Q
col6=[1.0, 2.0, 3.0]) # Q
selected_x = 'col1'
selected_y = utils.select_y(data, selected_x, ['N', 'T', 'Q', 'O'])
assert selected_y == 'col2'
def test_select_y():
def _check(d, expected):
x = 'col1'
y = utils.select_y(d, x)
assert y == expected
data = dict(col1=[1.0, 2.0, 3.0], # Chosen X
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
col5=[1.0, 2.0, 3.0]) # Q
_check(data, 'col5')
data = dict(col1=[1.0, 2.0, 3.0], # Chosen X
col2=['A', 'B', 'C'], # N
col3=pd.date_range('2012', periods=3, freq='A')) # T
_check(data, 'col2')
data = dict(col1=[1.0, 2.0, 3.0], # Chosen X
col2=pd.date_range('2012', periods=3, freq='A')) # T
_check(data, 'col2')
# No data
assert utils.select_y(None, "something") is None
# Len < 2
assert utils.select_y(dict(col1=[1.0, 2.0, 3.0]), "something") is None
# No x
assert utils.select_y(df, None) is None
# Custom order
data = dict(col1=[1.0, 2.0, 3.0], # Chosen X
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
col5=[1.0, 2.0, 3.0], # Q
col6=[1.0, 2.0, 3.0]) # Q
selected_x = 'col1'
selected_y = utils.select_y(data, selected_x, ['N', 'T', 'Q', 'O'])
assert selected_y == 'col2'
def _check(d, expected):
x = 'col1'
y = utils.select_y(d, x)
assert y == expected
def _check(d, expected):
x = 'col1'
y = utils.select_y(d, x)
assert y == expected
