def test_set_level(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_level(NEGATIVE_NUMBER)
self.assertEqual(battery_test.level, 0)
# check number positive the limit
battery_test.set_level(POSITIVE_NUMBER)
self.assertEqual(battery_test.get_level(), POSITIVE_NUMBER)
# check number positive not in the limit
battery_test.set_level(101)
self.assertEqual(battery_test.get_level(), 100)
# check float number positive not in the limit
battery_test.set_level(10.15)
self.assertEqual(battery_test.get_level(), 10)
# check ZERO
battery_test.set_level(0)
self.assertEqual(battery_test.get_level(), 0)
#check char that non number
battery_test.set_level(CHARACTER)
self.assertEqual(battery_test.get_level(), 0)
#check char that number
battery_test.set_level(CHAR_NUMBER)
self.assertEqual(battery_test.get_level(), int(CHAR_NUMBER))
sitl.stop()
