@@ -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+
271385class 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-
517517class TestValueRegexFilter (unittest2 .TestCase ):
518518
519519 def _getTargetClass (self ):
0 commit comments