|
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