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()