def test_get_assert_function_pandas_series(): try: import pandas as pd import pandas.util.testing except ImportError: return expected = pd.util.testing.assert_series_equal actual = ipytest.get_assert_function(pd.Series(), None) ipytest.assert_equals(actual, expected) actual = ipytest.get_assert_function(None, pd.Series()) ipytest.assert_equals(actual, expected) actual = ipytest.get_assert_function(pd.Series(), pd.Series()) ipytest.assert_equals(actual, expected)
def test_get_assert_function_pandas_frame(): try: import pandas as pd import pandas.util.testing as pdt except ImportError: return expected = pdt.assert_frame_equal actual = ipytest.get_assert_function(pd.DataFrame(), None) ipytest.assert_equals(actual, expected) actual = ipytest.get_assert_function(None, pd.DataFrame()) ipytest.assert_equals(actual, expected) actual = ipytest.get_assert_function(pd.DataFrame(), pd.DataFrame()) ipytest.assert_equals(actual, expected)
def test_nonequal_fails(self): with self.assertRaises(AssertionError): ipytest.assert_equals(1, 2)