diff --git a/wlauto/instrumentation/netstats/__init__.py b/wlauto/instrumentation/netstats/__init__.py index 48967101..78de2e92 100644 --- a/wlauto/instrumentation/netstats/__init__.py +++ b/wlauto/instrumentation/netstats/__init__.py @@ -9,7 +9,7 @@ from itertools import izip_longest from wlauto import Instrument, Parameter from wlauto import ApkFile -from wlauto.exceptions import DeviceError, HostError +from wlauto.exceptions import InstrumentError, HostError from wlauto.utils.android import ApkInfo from wlauto.utils.types import list_of_strings @@ -160,7 +160,7 @@ class NetstatsInstrument(Instrument): def initialize(self, context): if self.device.platform != 'android': - raise DeviceError('nestats instrument only supports on Android devices.') + raise InstrumentError('nestats instrument is only supported on Android devices.') apk = context.resolver.get(ApkFile(self)) self.collector = NetstatsCollector(self.device, apk) # pylint: disable=attribute-defined-outside-init self.collector.setup(force=self.force_reinstall) diff --git a/wlauto/instrumentation/servo_power_monitors/__init__.py b/wlauto/instrumentation/servo_power_monitors/__init__.py index 4cf94861..49e09b53 100644 --- a/wlauto/instrumentation/servo_power_monitors/__init__.py +++ b/wlauto/instrumentation/servo_power_monitors/__init__.py @@ -117,6 +117,10 @@ class ServoPowerMonitor(Instrument): self.poller = None self.data = None self.stopped = True + + if self.device.platform != "chromeos": + raise InstrumentError("servo_power instrument only supports Chrome OS devices.") + if not self.labels: self.labels = ["PORT_{}".format(channel) for channel, _ in enumerate(self.power_domains)]