Skip to content

Commit f284923

Browse files
committed
Merge pull request #1335 from dhermes/re-order-TestColumnRangeFilter
Moving Bigtable TestColumnRangeFilter within unit test file.
2 parents 138fd2c + b82bb02 commit f284923

1 file changed

Lines changed: 114 additions & 114 deletions

File tree

gcloud/bigtable/test_row.py

Lines changed: 114 additions & 114 deletions
Original file line numberDiff line numberDiff line change
@@ -268,6 +268,120 @@ def test_to_pb(self):
268268
self.assertEqual(pb_val, expected_pb)
269269

270270

271+
class TestTimestampRange(unittest2.TestCase):
272+
273+
def _getTargetClass(self):
274+
from gcloud.bigtable.row import TimestampRange
275+
return TimestampRange
276+
277+
def _makeOne(self, *args, **kwargs):
278+
return self._getTargetClass()(*args, **kwargs)
279+
280+
def test_constructor(self):
281+
start = object()
282+
end = object()
283+
time_range = self._makeOne(start=start, end=end)
284+
self.assertTrue(time_range.start is start)
285+
self.assertTrue(time_range.end is end)
286+
287+
def test___eq__(self):
288+
start = object()
289+
end = object()
290+
time_range1 = self._makeOne(start=start, end=end)
291+
time_range2 = self._makeOne(start=start, end=end)
292+
self.assertEqual(time_range1, time_range2)
293+
294+
def test___eq__type_differ(self):
295+
start = object()
296+
end = object()
297+
time_range1 = self._makeOne(start=start, end=end)
298+
time_range2 = object()
299+
self.assertNotEqual(time_range1, time_range2)
300+
301+
def test___ne__same_value(self):
302+
start = object()
303+
end = object()
304+
time_range1 = self._makeOne(start=start, end=end)
305+
time_range2 = self._makeOne(start=start, end=end)
306+
comparison_val = (time_range1 != time_range2)
307+
self.assertFalse(comparison_val)
308+
309+
def _to_pb_helper(self, start_micros=None, end_micros=None):
310+
import datetime
311+
from gcloud._helpers import _EPOCH
312+
from gcloud.bigtable._generated import bigtable_data_pb2 as data_pb2
313+
314+
pb_kwargs = {}
315+
316+
start = None
317+
if start_micros is not None:
318+
start = _EPOCH + datetime.timedelta(microseconds=start_micros)
319+
pb_kwargs['start_timestamp_micros'] = start_micros
320+
end = None
321+
if end_micros is not None:
322+
end = _EPOCH + datetime.timedelta(microseconds=end_micros)
323+
pb_kwargs['end_timestamp_micros'] = end_micros
324+
time_range = self._makeOne(start=start, end=end)
325+
326+
expected_pb = data_pb2.TimestampRange(**pb_kwargs)
327+
self.assertEqual(time_range.to_pb(), expected_pb)
328+
329+
def test_to_pb(self):
330+
# Makes sure already milliseconds granularity
331+
start_micros = 30871000
332+
end_micros = 12939371000
333+
self._to_pb_helper(start_micros=start_micros,
334+
end_micros=end_micros)
335+
336+
def test_to_pb_start_only(self):
337+
# Makes sure already milliseconds granularity
338+
start_micros = 30871000
339+
self._to_pb_helper(start_micros=start_micros)
340+
341+
def test_to_pb_end_only(self):
342+
# Makes sure already milliseconds granularity
343+
end_micros = 12939371000
344+
self._to_pb_helper(end_micros=end_micros)
345+
346+
347+
class TestTimestampRangeFilter(unittest2.TestCase):
348+
349+
def _getTargetClass(self):
350+
from gcloud.bigtable.row import TimestampRangeFilter
351+
return TimestampRangeFilter
352+
353+
def _makeOne(self, *args, **kwargs):
354+
return self._getTargetClass()(*args, **kwargs)
355+
356+
def test_constructor(self):
357+
range_ = object()
358+
row_filter = self._makeOne(range_)
359+
self.assertTrue(row_filter.range_ is range_)
360+
361+
def test___eq__type_differ(self):
362+
range_ = object()
363+
row_filter1 = self._makeOne(range_)
364+
row_filter2 = object()
365+
self.assertNotEqual(row_filter1, row_filter2)
366+
367+
def test___eq__same_value(self):
368+
range_ = object()
369+
row_filter1 = self._makeOne(range_)
370+
row_filter2 = self._makeOne(range_)
371+
self.assertEqual(row_filter1, row_filter2)
372+
373+
def test_to_pb(self):
374+
from gcloud.bigtable._generated import bigtable_data_pb2 as data_pb2
375+
from gcloud.bigtable.row import TimestampRange
376+
377+
range_ = TimestampRange()
378+
row_filter = self._makeOne(range_)
379+
pb_val = row_filter.to_pb()
380+
expected_pb = data_pb2.RowFilter(
381+
timestamp_range_filter=data_pb2.TimestampRange())
382+
self.assertEqual(pb_val, expected_pb)
383+
384+
271385
class TestColumnRangeFilter(unittest2.TestCase):
272386

273387
def _getTargetClass(self):
@@ -400,120 +514,6 @@ def test_to_pb_exclusive_end(self):
400514
self.assertEqual(row_filter.to_pb(), expected_pb)
401515

402516

403-
class TestTimestampRange(unittest2.TestCase):
404-
405-
def _getTargetClass(self):
406-
from gcloud.bigtable.row import TimestampRange
407-
return TimestampRange
408-
409-
def _makeOne(self, *args, **kwargs):
410-
return self._getTargetClass()(*args, **kwargs)
411-
412-
def test_constructor(self):
413-
start = object()
414-
end = object()
415-
time_range = self._makeOne(start=start, end=end)
416-
self.assertTrue(time_range.start is start)
417-
self.assertTrue(time_range.end is end)
418-
419-
def test___eq__(self):
420-
start = object()
421-
end = object()
422-
time_range1 = self._makeOne(start=start, end=end)
423-
time_range2 = self._makeOne(start=start, end=end)
424-
self.assertEqual(time_range1, time_range2)
425-
426-
def test___eq__type_differ(self):
427-
start = object()
428-
end = object()
429-
time_range1 = self._makeOne(start=start, end=end)
430-
time_range2 = object()
431-
self.assertNotEqual(time_range1, time_range2)
432-
433-
def test___ne__same_value(self):
434-
start = object()
435-
end = object()
436-
time_range1 = self._makeOne(start=start, end=end)
437-
time_range2 = self._makeOne(start=start, end=end)
438-
comparison_val = (time_range1 != time_range2)
439-
self.assertFalse(comparison_val)
440-
441-
def _to_pb_helper(self, start_micros=None, end_micros=None):
442-
import datetime
443-
from gcloud._helpers import _EPOCH
444-
from gcloud.bigtable._generated import bigtable_data_pb2 as data_pb2
445-
446-
pb_kwargs = {}
447-
448-
start = None
449-
if start_micros is not None:
450-
start = _EPOCH + datetime.timedelta(microseconds=start_micros)
451-
pb_kwargs['start_timestamp_micros'] = start_micros
452-
end = None
453-
if end_micros is not None:
454-
end = _EPOCH + datetime.timedelta(microseconds=end_micros)
455-
pb_kwargs['end_timestamp_micros'] = end_micros
456-
time_range = self._makeOne(start=start, end=end)
457-
458-
expected_pb = data_pb2.TimestampRange(**pb_kwargs)
459-
self.assertEqual(time_range.to_pb(), expected_pb)
460-
461-
def test_to_pb(self):
462-
# Makes sure already milliseconds granularity
463-
start_micros = 30871000
464-
end_micros = 12939371000
465-
self._to_pb_helper(start_micros=start_micros,
466-
end_micros=end_micros)
467-
468-
def test_to_pb_start_only(self):
469-
# Makes sure already milliseconds granularity
470-
start_micros = 30871000
471-
self._to_pb_helper(start_micros=start_micros)
472-
473-
def test_to_pb_end_only(self):
474-
# Makes sure already milliseconds granularity
475-
end_micros = 12939371000
476-
self._to_pb_helper(end_micros=end_micros)
477-
478-
479-
class TestTimestampRangeFilter(unittest2.TestCase):
480-
481-
def _getTargetClass(self):
482-
from gcloud.bigtable.row import TimestampRangeFilter
483-
return TimestampRangeFilter
484-
485-
def _makeOne(self, *args, **kwargs):
486-
return self._getTargetClass()(*args, **kwargs)
487-
488-
def test_constructor(self):
489-
range_ = object()
490-
row_filter = self._makeOne(range_)
491-
self.assertTrue(row_filter.range_ is range_)
492-
493-
def test___eq__type_differ(self):
494-
range_ = object()
495-
row_filter1 = self._makeOne(range_)
496-
row_filter2 = object()
497-
self.assertNotEqual(row_filter1, row_filter2)
498-
499-
def test___eq__same_value(self):
500-
range_ = object()
501-
row_filter1 = self._makeOne(range_)
502-
row_filter2 = self._makeOne(range_)
503-
self.assertEqual(row_filter1, row_filter2)
504-
505-
def test_to_pb(self):
506-
from gcloud.bigtable._generated import bigtable_data_pb2 as data_pb2
507-
from gcloud.bigtable.row import TimestampRange
508-
509-
range_ = TimestampRange()
510-
row_filter = self._makeOne(range_)
511-
pb_val = row_filter.to_pb()
512-
expected_pb = data_pb2.RowFilter(
513-
timestamp_range_filter=data_pb2.TimestampRange())
514-
self.assertEqual(pb_val, expected_pb)
515-
516-
517517
class TestValueRegexFilter(unittest2.TestCase):
518518

519519
def _getTargetClass(self):

0 commit comments

Comments
 (0)