|
6 | 6 | import numpy as np
|
7 | 7 |
|
8 | 8 | from pandas._libs.tslibs import Timedelta, Timestamp
|
9 |
| -from pandas.compat import DeepChainMap, string_types, u |
| 9 | +from pandas.compat import DeepChainMap, string_types |
10 | 10 |
|
11 | 11 | from pandas.core.dtypes.common import is_list_like
|
12 | 12 |
|
@@ -182,43 +182,43 @@ def stringify(value):
|
182 | 182 |
|
183 | 183 | kind = _ensure_decoded(self.kind)
|
184 | 184 | meta = _ensure_decoded(self.meta)
|
185 |
| - if kind == u('datetime64') or kind == u('datetime'): |
| 185 | + if kind == 'datetime64' or kind == 'datetime': |
186 | 186 | if isinstance(v, (int, float)):
|
187 | 187 | v = stringify(v)
|
188 | 188 | v = _ensure_decoded(v)
|
189 | 189 | v = Timestamp(v)
|
190 | 190 | if v.tz is not None:
|
191 | 191 | v = v.tz_convert('UTC')
|
192 | 192 | return TermValue(v, v.value, kind)
|
193 |
| - elif kind == u('timedelta64') or kind == u('timedelta'): |
| 193 | + elif kind == 'timedelta64' or kind == 'timedelta': |
194 | 194 | v = Timedelta(v, unit='s').value
|
195 | 195 | return TermValue(int(v), v, kind)
|
196 |
| - elif meta == u('category'): |
| 196 | + elif meta == 'category': |
197 | 197 | metadata = com.values_from_object(self.metadata)
|
198 | 198 | result = metadata.searchsorted(v, side='left')
|
199 | 199 |
|
200 | 200 | # result returns 0 if v is first element or if v is not in metadata
|
201 | 201 | # check that metadata contains v
|
202 | 202 | if not result and v not in metadata:
|
203 | 203 | result = -1
|
204 |
| - return TermValue(result, result, u('integer')) |
205 |
| - elif kind == u('integer'): |
| 204 | + return TermValue(result, result, 'integer') |
| 205 | + elif kind == 'integer': |
206 | 206 | v = int(float(v))
|
207 | 207 | return TermValue(v, v, kind)
|
208 |
| - elif kind == u('float'): |
| 208 | + elif kind == 'float': |
209 | 209 | v = float(v)
|
210 | 210 | return TermValue(v, v, kind)
|
211 |
| - elif kind == u('bool'): |
| 211 | + elif kind == 'bool': |
212 | 212 | if isinstance(v, string_types):
|
213 |
| - v = not v.strip().lower() in [u('false'), u('f'), u('no'), |
214 |
| - u('n'), u('none'), u('0'), |
215 |
| - u('[]'), u('{}'), u('')] |
| 213 | + v = not v.strip().lower() in ['false', 'f', 'no', |
| 214 | + 'n', 'none', '0', |
| 215 | + '[]', '{}', ''] |
216 | 216 | else:
|
217 | 217 | v = bool(v)
|
218 | 218 | return TermValue(v, v, kind)
|
219 | 219 | elif isinstance(v, string_types):
|
220 | 220 | # string quoting
|
221 |
| - return TermValue(v, stringify(v), u('string')) |
| 221 | + return TermValue(v, stringify(v), 'string') |
222 | 222 | else:
|
223 | 223 | raise TypeError("Cannot compare {v} of type {typ} to {kind} column"
|
224 | 224 | .format(v=v, typ=type(v), kind=kind))
|
|
0 commit comments