ó ÎYc@@sēdZddlmZddlZddlmZmZmZddlm Z m Z d„Z d„Z d„Z d „Zd „Zd „Zd „Zd „Zd„Zd„ZdS(s. This is the Bokeh charts testing interface. i(tabsolute_importN(t ColorAttrtDashAttrt MarkerAttr(tChartDataSourcetNumericalColumnsAssignercC@sLtj|jŒ}t|jƒdks-t‚t|jƒdksHt‚dS(s:Test creating chart data source from array-like list data.iiN(Rt from_datat list_datatlentcolumnstAssertionErrortindex(t test_datatds((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_listscC@sLtj|jŒ}t|jƒdks-t‚t|jƒdksHt‚dS(s;Test creating chart data source from array-like numpy data.iiN(RRt array_dataRR R R (R R ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_array#scC@sē|jjƒ}tj|ƒ}t|jƒdks9t‚t|jƒdksTt‚t|ƒ}t|jƒdks{t‚t|jƒdks–t‚|j|jƒsŪt‚dS(s8Test creating chart data source from existing dataframe.iiN( tpd_datatcopyRRRR R R tequals(R RR ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_pandas*s cC@sLtj|jƒ}t|jƒdks-t‚t|jƒdksHt‚dS(s4Test creating chart data source from dict of arrays.iiN(RRt dict_dataRR R R (R R ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_dict:scC@sLtj|jƒ}t|jƒdks-t‚t|jƒdksHt‚dS(s4Test creating chart data source from array of dicts.iiN(RRt records_dataRR R R (R R ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_recordsAsc C@sþxũtttfD]æ}i|d|jdddddddgƒd6}i|d|jdd dddddgƒd6}td|jƒ}t|j|ƒ}t|ƒd ks·t‚td|jƒ}t|j|ƒ}t|ƒd kst‚qWdS( NtdfR tcyltattrnamettesttiterabletredtbluetoriginii (RR ( RRRt auto_dataRtlisttgroupbyRR (R tclstsingle_col_spectmulti_col_specR tgroups((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_groupbyHscC@sltj|jŒ}y5g|jD]#}g|D]}||^q,^qWntk rgtjdƒnXdS(Ns8Required dimension not correctly set by ChartDataSource.(RRRt_required_dimstKeyErrortpytesttfail(R R treq_dimstdim((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyttest_derived_selections[s 5 cC@s5tjd|jƒ}|dddgks1t‚dS(s9List of lists for dimension results in column derivation.tytatbN(RRRR (R R ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyttest_derived_cols_from_lists_kwcscC@s8tj|jddƒ}|dddgks4t‚dS(s9List of lists for dimension results in column derivation.tdimsR0txR1R2N(R0R5(RRRR (R R ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyttest_derived_cols_from_listsiscC@s>tj|jdddtƒ}|dddgks:t‚dS( s(Wide dataframe columns assigned to y dimension for the chart. This tests using the NumericalColumnsAssigner, which is used for charts that prefer behavior of expecting series-like data. Typically the x axis is used as an index, then the y axis will be assigned multiple columns. R4R5R0tcolumn_assignertcol1tcol2N(R5R0(RRRRR (R R ((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyt test_custom_derived_cols_from_dfos (t__doc__t __future__RR+tbkcharts.attributesRRRtbkcharts.data_sourceRRRRRRRR(R/R3R6R:(((s>lib/python2.7/site-packages/bkcharts/tests/test_data_source.pyts