def test_resort_dataset_no_rule(self): ''' Test if dataset resturns as is if no order rule is sent ''' self.assertEqual( BaseModel.resort_dataset( SAMPLE_DATAFRAME, ["col_1", "col_2"] ).to_dict('split')['data'], [['a', 1, 1], ['b', 2, 2], ['c', 0, 0], ['d', 3, 3]] )
def test_resort_dataset_desc(self): ''' Test if dataset is ordered descending if all args are descending ''' self.assertEqual( BaseModel.resort_dataset( SAMPLE_DATAFRAME, ["-col_1", "-col_2", "-col_3"] ).to_dict('split')['data'], [['d', 3, 3], ['c', 0, 0], ['b', 2, 2], ['a', 1, 1]] )