KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use test_multiindex method in pandera

Best Python code snippet using pandera_python

test_from_pandas.py

Source:test_from_pandas.py

...46            [2, 'b'],47            [3, 'c'],48        ]49        self.assertEqual(list(reader), expected)50    def test_multiindex(self):51        index_values = [('x', 'one'), ('x', 'two'), ('y', 'three')]52        index = pandas.MultiIndex.from_tuples(index_values, names=['A', 'B'])53        self.df.index = index54        reader = _from_pandas(self.df)55        expected = [56            ['A', 'B', 'col1', 'col2'],57            ['x', 'one', 1, 'a'],58            ['x', 'two', 2, 'b'],59            ['y', 'three', 3, 'c'],60        ]61        self.assertEqual(list(reader), expected)62        reader = _from_pandas(self.df, index=False)63        expected = [64            ['col1', 'col2'],65            [1, 'a'],66            [2, 'b'],67            [3, 'c'],68        ]69        self.assertEqual(list(reader), expected)70@unittest.skipIf(not pandas, 'pandas not found')71class TestSeries(unittest.TestCase):72    def test_automatic_indexing(self):73        s = pandas.Series(['a', 'b', 'c'], name='mycolumn')74        reader = _from_pandas(s)  # <- Includes index by default.75        expected = [76            [None, 'mycolumn'],77            [0, 'a'],78            [1, 'b'],79            [2, 'c'],80        ]81        self.assertEqual(list(reader), expected)82        reader = _from_pandas(s, index=False)  # <- Omits index.83        expected = [84            ['mycolumn'],85            ['a'],86            ['b'],87            ['c'],88        ]89        self.assertEqual(list(reader), expected)90    def test_simple_index(self):91        s = pandas.Series(92            data=['a', 'b', 'c'],93            index=pandas.Index(['x', 'y', 'z'], name='myindex'),94            name='mycolumn',95        )96        reader = _from_pandas(s)97        expected = [98            ['myindex', 'mycolumn'],99            ['x', 'a'],100            ['y', 'b'],101            ['z', 'c'],102        ]103        self.assertEqual(list(reader), expected)104        reader = _from_pandas(s, index=False)105        expected = [106            ['mycolumn'],107            ['a'],108            ['b'],109            ['c'],110        ]111        self.assertEqual(list(reader), expected)112    def test_multiindex(self):113        multiindex = [('x', 'one'), ('x', 'two'), ('y', 'three')]114        multiindex = pandas.MultiIndex.from_tuples(multiindex, names=['A', 'B'])115        s = pandas.Series(116            data=['a', 'b', 'c'],117            index=multiindex,118            name='mycolumn',119        )120        reader = _from_pandas(s)121        expected = [122            ['A', 'B', 'mycolumn'],123            ['x', 'one', 'a'],124            ['x', 'two', 'b'],125            ['y', 'three', 'c'],126        ]127        self.assertEqual(list(reader), expected)128        reader = _from_pandas(s, index=False)129        expected = [130            ['mycolumn'],131            ['a'],132            ['b'],133            ['c'],134        ]135        self.assertEqual(list(reader), expected)136@unittest.skipIf(not pandas, 'pandas not found')137class TestIndex(unittest.TestCase):138    def test_simple_index(self):139        index = pandas.Index(['x', 'y', 'z'], name='myindex')140        reader = _from_pandas(index)141        expected = [142            ['myindex'],143            ['x'],144            ['y'],145            ['z'],146        ]147        self.assertEqual(list(reader), expected)148        reader = _from_pandas(index, index=False)149        expected = [150            ['myindex'],151            ['x'],152            ['y'],153            ['z'],154        ]155        msg = 'The `index` arg should be ingored when obj is a pandas Index.'156        self.assertEqual(list(reader), expected, msg=msg)157    def test_multiindex(self):158        multiindex = [('x', 'one'), ('x', 'two'), ('y', 'three')]159        multiindex = pandas.MultiIndex.from_tuples(multiindex, names=['A', 'B'])160        reader = _from_pandas(multiindex)161        expected = [162            ['A', 'B'],163            ['x', 'one'],164            ['x', 'two'],165            ['y', 'three'],166        ]167        self.assertEqual(list(reader), expected)168        reader = _from_pandas(multiindex, index=False)169        expected = [170            ['A', 'B'],171            ['x', 'one'],...

test_json_table_schema.pyi

Source:test_json_table_schema.pyi

...15    def test_series(self) -> None:16        ...17    def test_series_unnamed(self) -> None:18        ...19    def test_multiindex(self) -> None:20        ...21class TestTableSchemaType:22    def test_as_json_table_type_int_data(self, int_type: Any) -> None:23        ...24    def test_as_json_table_type_float_data(self, float_type: Any) -> None:25        ...26    def test_as_json_table_type_bool_data(self, bool_type: Any) -> None:27        ...28    def test_as_json_table_type_date_data(self, date_data: Any) -> None:29        ...30    def test_as_json_table_type_string_data(self, str_data: Any) -> None:31        ...32    def test_as_json_table_type_categorical_data(self, cat_data: Any) -> None:33        ...34    def test_as_json_table_type_int_dtypes(self, int_dtype: Any) -> None:35        ...36    def test_as_json_table_type_float_dtypes(self, float_dtype: Any) -> None:37        ...38    def test_as_json_table_type_bool_dtypes(self, bool_dtype: Any) -> None:39        ...40    def test_as_json_table_type_date_dtypes(self, date_dtype: Any) -> None:41        ...42    def test_as_json_table_type_timedelta_dtypes(self, td_dtype: Any) -> None:43        ...44    def test_as_json_table_type_string_dtypes(self, str_dtype: Any) -> None:45        ...46    def test_as_json_table_type_categorical_dtypes(self) -> None:47        ...48class TestTableOrient:49    df: Any = ...50    def setup_method(self, method: Any) -> None:51        ...52    def test_build_series(self) -> None:53        ...54    def test_to_json(self) -> None:55        ...56    def test_to_json_float_index(self) -> None:57        ...58    def test_to_json_period_index(self) -> None:59        ...60    def test_to_json_categorical_index(self) -> None:61        ...62    def test_date_format_raises(self) -> None:63        ...64    def test_convert_pandas_type_to_json_field_int(self, kind: Any) -> None:65        ...66    def test_convert_pandas_type_to_json_field_float(self, kind: Any) -> None:67        ...68    def test_convert_pandas_type_to_json_field_datetime(self, dt_args: Any, extra_exp: Any, wrapper: Any) -> None:69        ...70    def test_convert_pandas_type_to_json_period_range(self) -> None:71        ...72    def test_convert_pandas_type_to_json_field_categorical(self, kind: Any, ordered: Any) -> None:73        ...74    def test_convert_json_field_to_pandas_type(self, inp: Any, exp: Any) -> None:75        ...76    def test_convert_json_field_to_pandas_type_raises(self, inp: Any) -> None:77        ...78    def test_categorical(self) -> None:79        ...80    def test_set_names_unset(self, idx: Any, nm: Any, prop: Any) -> None:81        ...82    def test_warns_non_roundtrippable_names(self, idx: Any) -> None:83        ...84    def test_timestamp_in_columns(self) -> None:85        ...86    def test_overlapping_names(self, case: Any) -> None:87        ...88    def test_mi_falsey_name(self) -> None:89        ...90class TestTableOrientReader:91    def test_read_json_table_orient(self, index_nm: Any, vals: Any, recwarn: Any) -> None:92        ...93    def test_read_json_table_orient_raises(self, index_nm: Any, vals: Any, recwarn: Any) -> None:94        ...95    def test_comprehensive(self) -> None:96        ...97    def test_multiindex(self, index_names: Any) -> None:98        ...99    def test_empty_frame_roundtrip(self) -> None:...

test_compare.py

Source:test_compare.py

...67        "Flavour ID": [1, 2, None, None, None, None],68        "Flavour Description": ["Fuzzy", "Bubbly", "Fuzzy", "Fuzzy", "Fuzzy", "Sugary"],69    })70    return df_1, df_271def test_multiindex():72    df_1, df_2 = sample_multiindex_dfs()73    diff = compare.diff_pd(df_1, df_2, ["ID", "Flavour ID"])74    run_assertion(diff)75def test_multiindex_excel():76    xlsx_1, xlsx_2 = sample_xlsx(*sample_multiindex_dfs())77    diff_io = io.BytesIO()78    compare.compare_excel(xlsx_1, xlsx_2, diff_io, "Sheet1", ["ID", "Flavour ID"])79    diff = pd.read_excel(diff_io, sheet_name=None)80    run_assertion(diff)81    82def test_no_diffs():83    df_1, _ = sample_multiindex_dfs()84    diff = compare.diff_pd(df_1, df_1, ["ID", "Flavour ID"])85    assert not diff86    print("OK.")87if __name__ == '__main__':88    test_multiindex()89    test_multiindex_excel()90    test_single_index()91    test_single_index_excel()92    test_parser()...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.