| from sympy import S, Basic, Set, ordered, sympify | |
| from sympy.sets import FiniteSet as SympyFiniteSet | |
| from sympy.core.parameters import global_parameters | |
| class FiniteSet(SympyFiniteSet): | |
| """ | |
| FiniteSet which keeps the _unsorted_args attribute, only available till the first evaluation | |
| """ | |
| def __new__(cls, *args, **kwargs): | |
| evaluate = kwargs.get('evaluate', global_parameters.evaluate) | |
| if evaluate: | |
| args = list(map(sympify, args)) | |
| if len(args) == 0: | |
| return S.EmptySet | |
| else: | |
| args = list(map(sympify, args)) | |
| unsorted_args = args | |
| # keep the form of the first canonical arg | |
| dargs = {} | |
| for i in reversed(list(ordered(args))): | |
| if i.is_Symbol: | |
| dargs[i] = i | |
| else: | |
| try: | |
| dargs[i.as_dummy()] = i | |
| except TypeError: | |
| # e.g. i = class without args like `Interval` | |
| dargs[i] = i | |
| _args_set = set(dargs.values()) | |
| args = list(ordered(_args_set, Set._infimum_key)) | |
| obj = Basic.__new__(cls, *args) | |
| obj._args_set = _args_set | |
| obj._unsorted_args = unsorted_args | |
| return obj | |