Fix for Issue #50.

Author: jamesleesaunders

xbee/base.py

@@ -181,6 +181,9 @@ def _build_command(self, cmd, **kwargs):
             try:
                 # Read this field's name from the function arguments dict
                 data = kwargs[field['name']]
+                if isinstance(data, str):
+                    data = str.encode(data)
+
             except KeyError:
                 # Data wasn't given
                 # Only a problem if the field has a specific length
@@ -205,9 +208,8 @@ def _build_command(self, cmd, **kwargs):
                     "The data provided for '%s' was not %d bytes long"\
                     % (field['name'], field['len']))
 
-            # Add the data to the packet, if it has been specified
-            # Otherwise, the parameter was of variable length, and not
-            #  given
+            # Add the data to the packet, if it has been specified.
+            # Otherwise, the parameter was of variable length, and not given.
             if data:
                 packet += data
 
@@ -393,7 +395,6 @@ def send(self, cmd, **kwargs):
         # Pass through the keyword arguments
         self._write(self._build_command(cmd, **kwargs))
 
-
     def wait_read_frame(self):
         """
         wait_read_frame: None -> frame info dictionary

xbee/tests/test_zigbee.py

@@ -8,6 +8,7 @@
 """
 import unittest
 from xbee.zigbee import ZigBee
+from xbee.tests.Fake import Serial
 
 class TestZigBee(unittest.TestCase):
     """
@@ -17,6 +18,17 @@ class TestZigBee(unittest.TestCase):
     def setUp(self):
         self.zigbee = ZigBee(None)
 
+    def test_send(self):
+        """
+        Test send() with AT command.
+        """
+        device = Serial()
+        xbee = ZigBee(device)
+        xbee.send('at', command='MY')
+        result = device.get_data_written()
+        expected = b'~\x00\x04\x08\x01MYP'
+        self.assertEqual(result, expected)
+
     def test_null_terminated_field(self):
         """
         Packets with null-terminated fields