(capsys)
| 53 | not haveIPython, reason="Can't run variable tests without IPython console" |
| 54 | ) |
| 55 | def test_dataframe_info(capsys): |
| 56 | from IPython import get_ipython |
| 57 | |
| 58 | # Setup some different types |
| 59 | get_ipython().run_cell( |
| 60 | """ |
| 61 | import pandas as pd |
| 62 | import numpy as np |
| 63 | ls = list([10, 20, 30, 40]) |
| 64 | df = pd.DataFrame(ls) |
| 65 | se = pd.Series(ls) |
| 66 | np1 = np.array(ls) |
| 67 | np2 = np.array([[1, 2, 3], [4, 5, 6]]) |
| 68 | dict1 = {'Name': 'Zara', 'Age': 7, 'Class': 'First'} |
| 69 | obj = {} |
| 70 | col = pd.Series(data=np.random.random_sample((7,))*100) |
| 71 | dfInit = {} |
| 72 | idx = pd.date_range('2007-01-01', periods=7, freq='M') |
| 73 | for i in range(30): |
| 74 | dfInit[i] = col |
| 75 | dfInit['idx'] = idx |
| 76 | df2 = pd.DataFrame(dfInit).set_index('idx') |
| 77 | df3 = df2.iloc[:, [0,1]] |
| 78 | se2 = df2.loc[df2.index[0], :] |
| 79 | """ |
| 80 | ) |
| 81 | vars = get_variables(capsys) |
| 82 | df = get_variable_value(vars, "df", capsys) |
| 83 | se = get_variable_value(vars, "se", capsys) |
| 84 | np = get_variable_value(vars, "np1", capsys) |
| 85 | np2 = get_variable_value(vars, "np2", capsys) |
| 86 | ls = get_variable_value(vars, "ls", capsys) |
| 87 | obj = get_variable_value(vars, "obj", capsys) |
| 88 | df3 = get_variable_value(vars, "df3", capsys) |
| 89 | se2 = get_variable_value(vars, "se2", capsys) |
| 90 | dict1 = get_variable_value(vars, "dict1", capsys) |
| 91 | assert df |
| 92 | assert se |
| 93 | assert np |
| 94 | assert ls |
| 95 | assert obj |
| 96 | assert df3 |
| 97 | assert se2 |
| 98 | assert dict1 |
| 99 | verify_dataframe_info(vars, "df", "index", capsys, True) |
| 100 | verify_dataframe_info(vars, "se", "index", capsys, True) |
| 101 | verify_dataframe_info(vars, "np1", "index", capsys, True) |
| 102 | verify_dataframe_info(vars, "ls", "index", capsys, True) |
| 103 | verify_dataframe_info(vars, "np2", "index", capsys, True) |
| 104 | verify_dataframe_info(vars, "obj", "index", capsys, False) |
| 105 | verify_dataframe_info(vars, "df3", "idx", capsys, True) |
| 106 | verify_dataframe_info(vars, "se2", "index", capsys, True) |
| 107 | verify_dataframe_info(vars, "df2", "idx", capsys, True) |
| 108 | verify_dataframe_info(vars, "dict1", "index", capsys, True) |
| 109 | |
| 110 | |
| 111 | def verify_dataframe_info(vars, name, indexColumn, capsys, hasInfo): |
nothing calls this directly
no test coverage detected