balance.testutil

class balance.testutil.BalanceTestCase(methodName='runTest')[source]
assertEqual(first: ndarray | DataFrame | Index | Series | Any, second: ndarray | DataFrame | Index | Series | Any, msg: Any = Ellipsis, **kwargs) None[source]

Check if first and second are equal. Uses np.testing.assert_array_equal for np.ndarray, _assert_frame_equal_lazy for pd.DataFrame, assert_series_equal for pd.DataFrame, _assert_index_equal_lazy for pd.Index, or unittest.TestCase.assertEqual otherwise.

Parameters:
  • first (Union[np.ndarray, pd.DataFrame, pd.Index, pd.Series]) – first element to compare.

  • second (Union[np.ndarray, pd.DataFrame, pd.Index, pd.Series]) – second element to compare.

  • msg (Any, optional) – The error message on failure.