def test_write_set(self): data_out = {1, 2, 3} data_path = os.path.join(self.dir.name, "1.tmp") utils.write_set(data_out, data_path) with open(data_path, "r") as fp: data_in = fp.read() self.assertEqual("\n".join((str(i) for i in sorted(data_out))) + "\n", data_in)
def command_make_difference(args): set1, set2 = utils.read_two_sets(args) utils.write_set(set1 - set2)
def command_make_union(args): result = set() for s in utils.read_many_sets(args): result |= s utils.write_set(result)
def command_make_intersection(args): result = set() for elem in utils.read_many_sets(args): result &= elem utils.write_set(result)