class DuckiecallNode(object): def __init__(self): self.node_name = 'duckiecall_node' # Setup the publisher and subscriber self.sub_num_of_quacks = rospy.Subscriber("~number_of_quacks", Int32, self.quacksCallback) self.pub_duckiecall = rospy.Publisher("~duckiecall", String, queue_size=1) # Setup the quacker self.quacker = Quacker() rospy.loginfo("[%s] has started", self.node_name) def quacksCallback(self, msg_quacks): msg_duckiecall = String() msg_duckiecall.data = self.quacker.get_quack_string(msg_quacks.data) self.pub_duckiecall.publish(msg_duckiecall)
def test_get_quack_string(self): quacker = Quacker("Quack!") msg = quacker.get_quack_string(3) self.assertEqual(msg, "Quack! Quack! Quack!")