def test_partial_anticlockwise_turn(self):
stroke_array = []
stroke_array.append([800, 2])
stroke_array.append([810, 1])
tsb = IRBatcher(stroke_array)
converted_ary = tsb.convert_stroke_array_to_display_array(1000)
self.assertEquals([0], converted_ary)
def test_anticlockwise_turn_spread_across_more_than_one_sample(self):
stroke_array = []
stroke_array.append([800, 3])
stroke_array.append([810, 2])
stroke_array.append([1110, 1])
tsb = IRBatcher(stroke_array)
converted_ary = tsb.convert_stroke_array_to_display_array(1000)
self.assertEquals([0, -1], converted_ary)
def test_turn_then_no_turn_then_turn(self):
stroke_array = []
tsb = IRBatcher(stroke_array)
stroke_array.append([800, 3])
stroke_array.append([810, 2])
stroke_array.append([820, 1])
stroke_array.append([2800, 3])
stroke_array.append([2810, 2])
stroke_array.append([2820, 1])
converted_ary = tsb.convert_stroke_array_to_display_array(1000)
self.assertEquals([-1, 0, -1], converted_ary)
def test_two_anticlockwise_turns(self):
stroke_array = []
stroke_array.append([800, 2])
stroke_array.append([810, 1])
stroke_array.append([820, 3])
stroke_array.append([830, 2])
stroke_array.append([840, 1])
stroke_array.append([850, 3])
tsb = IRBatcher(stroke_array)
converted_ary = tsb.convert_stroke_array_to_display_array(1000)
self.assertEquals([-2], converted_ary)
def test_multiple_full_anticlockwise_turns_spread_over_more_than_1_sample(
self):
stroke_array = []
stroke_array.append([800, 3])
stroke_array.append([810, 2])
stroke_array.append([820, 1])
stroke_array.append([1800, 3])
stroke_array.append([1810, 2])
stroke_array.append([1820, 1])
tsb = IRBatcher(stroke_array)
converted_ary = tsb.convert_stroke_array_to_display_array(1000)
self.assertEquals([-1, -1], converted_ary)
def __init__(self,
gst_img_path,
data_filename,
x,
trial_length,
display_sample,
layer_multiplier=1):
#super(PreBatchGhost, self).__init__(gst_img_path, data_filename, x, trial_length, display_sample)
#can't use super on an old style class
Ghost.__init__(self, gst_img_path, data_filename, x, layer_multiplier)
tb = TimestampBatcher(self.timestamp_ary)
self.batched_ts_ary = []
self.batched_ts_ary = tb.convert_ts_array_to_display_array(
trial_length, display_sample)
irtb = IRBatcher(self.ir_timestamp_ary)
self.batched_ir_ts_ary = []
self.batched_ir_ts_ary = irtb.convert_stroke_array_to_display_array(
display_sample)
self.sample_index = 0
self.stroke_index = 0
def test_empty_stroke_array(self):
stroke_array = []
tsb = IRBatcher(stroke_array)
converted_ary = tsb.convert_stroke_array_to_display_array(1000)
self.assertEquals([], converted_ary)