def test_set_volt(self):
sitl = SITL()
sitl.download('copter', '3.3', verbose=True)
sitl_args = ['-I0', '--model', 'quad', '--home=-35.363261,149.165230,584,353']
sitl.launch(sitl_args, await_ready=True, restart=True)
print "Connecting to vehicle on: 'tcp:127.0.0.1:5760'"
vehicle = connect('tcp:127.0.0.1:5760', wait_ready=True)
print " Battery: %s" % vehicle.battery
battery_test = Battery(vehicle.battery)
# check number negative the limit
battery_test.set_volt(NEGATIVE_NUMBER)
self.assertEqual(battery_test.volt, 0)
# check number positive the limit
battery_test.set_volt(POSITIVE_NUMBER)
self.assertEqual(battery_test.get_volt(), POSITIVE_NUMBER)
# check number positive the limit
battery_test.set_volt(12.4)
self.assertEqual(battery_test.get_volt(), 12.4)
# check ZERO
battery_test.set_volt(0)
self.assertEqual(battery_test.get_volt(), 0)
# check char that non number
battery_test.set_volt(CHARACTER)
self.assertEqual(battery_test.get_volt(), 0)
#check char that number
battery_test.set_volt(CHAR_NUMBER)
self.assertEqual(battery_test.get_volt(), int(CHAR_NUMBER))
sitl.stop()
