From d2bdb1303679b97c3de5ce2d2d1f2b62c9828a3a Mon Sep 17 00:00:00 2001
From: Bert Vermeulen <bert@biot.com>
Date: Mon, 23 Jun 2014 19:52:45 +0200
Subject: i2c: Generate proper exception for missing samplerate, and test for
 it.

Also remove some dead code.
---
 decoders/i2c/pd.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'decoders/i2c/pd.py')

diff --git a/decoders/i2c/pd.py b/decoders/i2c/pd.py
index 2d6e5c2..ced58cd 100644
--- a/decoders/i2c/pd.py
+++ b/decoders/i2c/pd.py
@@ -63,6 +63,9 @@ proto = {
     'DATA WRITE':      [9, 'Data write',    'DW'],
 }
 
+class SamplerateError(Exception):
+    pass
+
 class Decoder(srd.Decoder):
     api_version = 2
     id = 'i2c'
@@ -265,8 +268,8 @@ class Decoder(srd.Decoder):
         self.bits = []
 
     def decode(self, ss, es, data):
-        if self.samplerate is None:
-            raise Exception("Cannot decode without samplerate.")
+        if not self.samplerate:
+            raise SamplerateError("Cannot decode without samplerate.")
         for (self.samplenum, pins) in data:
 
             # Ignore identical samples early on (for performance reasons).
@@ -293,8 +296,6 @@ class Decoder(srd.Decoder):
             elif self.state == 'FIND ACK':
                 if self.is_data_bit(scl, sda):
                     self.get_ack(scl, sda)
-            else:
-                raise Exception('Invalid state: %s' % self.state)
 
             # Save current SDA/SCL values for the next round.
             self.oldscl, self.oldsda = scl, sda
-- 
cgit v1.2.3-70-g09d2