Skip to content

Commit

Permalink
Use print() function in both Python 2 and Python 3 in basic_localizat…
Browse files Browse the repository at this point in the history
…ion.py (#1011)
  • Loading branch information
cclauss committed Sep 22, 2020
1 parent bfda75e commit 3c16b37
Showing 1 changed file with 13 additions and 11 deletions.
24 changes: 13 additions & 11 deletions amcl/test/basic_localization.py
@@ -1,5 +1,7 @@
#!/usr/bin/env python

from __future__ import print_function

import sys
import time
from math import fmod, pi
Expand All @@ -25,10 +27,10 @@ def tf_cb(self, msg):
if t.header.frame_id == 'map':
self.tf = t.transform
(a_curr, a_diff) = self.compute_angle_diff()
print 'Curr:\t %16.6f %16.6f %16.6f' % (self.tf.translation.x, self.tf.translation.y, a_curr)
print 'Target:\t %16.6f %16.6f %16.6f' % (self.target_x, self.target_y, self.target_a)
print 'Diff:\t %16.6f %16.6f %16.6f' % (
abs(self.tf.translation.x - self.target_x), abs(self.tf.translation.y - self.target_y), a_diff)
print('Curr:\t %16.6f %16.6f %16.6f' % (self.tf.translation.x, self.tf.translation.y, a_curr))
print('Target:\t %16.6f %16.6f %16.6f' % (self.target_x, self.target_y, self.target_a))
print('Diff:\t %16.6f %16.6f %16.6f' % (
abs(self.tf.translation.x - self.target_x), abs(self.tf.translation.y - self.target_y), a_diff))

def compute_angle_diff(self):
rot = self.tf.rotation
Expand All @@ -47,28 +49,28 @@ def test_basic_localization(self):
target_time = float(sys.argv[7])

if global_localization == 1:
#print 'Waiting for service global_localization'
#print('Waiting for service global_localization')
rospy.wait_for_service('global_localization')
global_localization = rospy.ServiceProxy('global_localization', Empty)
global_localization()

rospy.init_node('test', anonymous=True)
while(rospy.rostime.get_time() == 0.0):
#print 'Waiting for initial time publication'
#print('Waiting for initial time publication')
time.sleep(0.1)
start_time = rospy.rostime.get_time()
# TODO: This should be replace by a pytf listener
rospy.Subscriber('/tf', TFMessage, self.tf_cb)

while (rospy.rostime.get_time() - start_time) < target_time:
#print 'Waiting for end time %.6f (current: %.6f)'%(target_time,(rospy.rostime.get_time() - start_time))
#print('Waiting for end time %.6f (current: %.6f)'%(target_time,(rospy.rostime.get_time() - start_time)))
time.sleep(0.1)

(a_curr, a_diff) = self.compute_angle_diff()
print 'Curr:\t %16.6f %16.6f %16.6f' % (self.tf.translation.x, self.tf.translation.y, a_curr)
print 'Target:\t %16.6f %16.6f %16.6f' % (self.target_x, self.target_y, self.target_a)
print 'Diff:\t %16.6f %16.6f %16.6f' % (
abs(self.tf.translation.x - self.target_x), abs(self.tf.translation.y - self.target_y), a_diff)
print('Curr:\t %16.6f %16.6f %16.6f' % (self.tf.translation.x, self.tf.translation.y, a_curr))
print('Target:\t %16.6f %16.6f %16.6f' % (self.target_x, self.target_y, self.target_a))
print('Diff:\t %16.6f %16.6f %16.6f' % (
abs(self.tf.translation.x - self.target_x), abs(self.tf.translation.y - self.target_y), a_diff))
self.assertNotEquals(self.tf, None)
self.assertTrue(abs(self.tf.translation.x - self.target_x) <= tolerance_d)
self.assertTrue(abs(self.tf.translation.y - self.target_y) <= tolerance_d)
Expand Down

0 comments on commit 3c16b37

Please sign in to comment.