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