def test_simple_disk_sort(self): self._clean_ds_tmp() actual = [(key, value) for key, value in disk_sort(RESPECTED, OUT_FILE, (0, 1))] print "ACTUAL: ", actual self.assertEqual(actual[0][0][0], "anymore") self.assertEqual(actual[1][0][1], 9) self.assertEqual(actual[2][0][0], "reallystuff") self.assertEqual(actual[3][1], ()) # tests secondary sorting
def test_positional_disk_sort(self): self._clean_ds_tmp() actual = [(key, value) for key, value in disk_sort(RESPECTED, OUT_FILE, [1])] print "ACTUAL: ", actual self.assertEqual(actual[0][0][0], "morestuff") self.assertEqual(actual[1][0][1], 15) self.assertEqual(actual[2][0][0], "stuff") self.assertEqual(actual[3][1], ()) self.assertEqual(actual[5][1], ())
def test_nulls(self): self._clean_ds_tmp() actual = [(key, value) for key, value in disk_sort(SOMENULLS, OUT_FILE, [0])] print "ACTUAL: ", actual self.assertEqual(actual[0][0][0], None)