def test_getting_bundle_data(self): '''Test bundling a signal and getting the data back''' block = Bundle(self.q_in, self.q_out) block.start() [self.q_in.put(i) for i in self.input + [LastEvent()]] block.join() bundled_data = self.q_out.get() self.assertEqual(len(bundled_data), 100) self.assertEqual(type(bundled_data), numpy.ndarray) values = bundled_data["Value"] self.assertTrue(all(values == 1)) tags = bundled_data["Tag"] [self.assertEquals(tags[i], i) for i in xrange(100)]
def test_getting_bundle_data( self ): '''Test bundling a signal and getting the data back''' block = Bundle( self.q_in, self.q_out ) block.start() [self.q_in.put( i ) for i in self.input + [LastEvent()]] block.join() bundled_data = self.q_out.get() self.assertEqual( len( bundled_data ), 100 ) self.assertEqual( type( bundled_data ), numpy.ndarray ) values = bundled_data["Value"] self.assertTrue( all( values == 1 ) ) tags = bundled_data["Tag"] [self.assertEquals( tags[i] , i ) for i in xrange( 100 ) ]
def test_bundle_full_signal(self): '''Test sending a basic integer tagged signal all at once''' bundle_size = 1000 #expected_output = [Event(value=1, tag=i + delay) for i in xrange(100)] block = Bundle(self.q_in, self.q_out, bundle_size) block.start() [self.q_in.put(i) for i in self.input + [LastEvent()]] block.join() actual_output = self.q_out.get() self.assertEqual(actual_output.size, 100) self.assertTrue(self.q_out.get().last)