From 7967e2f890567e75d50738ec95565aa9bebcf4e8 Mon Sep 17 00:00:00 2001
From: jay <jlaura@asu.edu>
Date: Thu, 27 Jul 2017 15:00:33 -0700
Subject: [PATCH] Fixes failing tests with control network API change.

---
 plio/io/io_controlnetwork.py            |  3 ++-
 plio/io/tests/test_io_controlnetwork.py | 26 +++++++++++++------------
 2 files changed, 16 insertions(+), 13 deletions(-)

diff --git a/plio/io/io_controlnetwork.py b/plio/io/io_controlnetwork.py
index 6af9b56..6efe11a 100644
--- a/plio/io/io_controlnetwork.py
+++ b/plio/io/io_controlnetwork.py
@@ -222,7 +222,8 @@ class IsisStore(object):
                 for attr, attrtype in self.measure_attrs:
                     if attr in g.columns:
                         setattr(measure_spec, attr, attrtype(m[attr]))
-
+                measure_spec.sample = m.x
+                measure_spec.line = m.y
                 measure_spec.type = 2
                 measure_iterable.append(measure_spec)
                 self.nmeasures += 1
diff --git a/plio/io/tests/test_io_controlnetwork.py b/plio/io/tests/test_io_controlnetwork.py
index bf77fb4..186967d 100644
--- a/plio/io/tests/test_io_controlnetwork.py
+++ b/plio/io/tests/test_io_controlnetwork.py
@@ -20,40 +20,42 @@ class TestWriteIsisControlNetwork(unittest.TestCase):
         cls.npts = 5
         serial_times = {295: '1971-07-31T01:24:11.754',
                         296: '1971-07-31T01:24:36.970'}
-        cls.serials = ['APOLLO15/METRIC/{}'.format(i) for i in serial_times.values()]
-        columns = ['point_id', 'point_type', 'serialnumber', 'measure_type', 'x', 'y', 'node_id']
+        cls.serials = {i:'APOLLO15/METRIC/{}'.format(j) for i, j in enumerate(serial_times.values())}
+        columns = ['point_id', 'point_type', 'serialnumber', 'measure_type', 'x', 'y', 'image_index']
+
+
 
         data = []
         for i in range(cls.npts):
             data.append((i, 2, cls.serials[0], 2, 0, 0, 0))
             data.append((i, 2, cls.serials[1], 2, 0, 0, 1))
 
-        dfs = [pd.DataFrame(data, columns=columns)]
+        df = pd.DataFrame(data, columns=columns)
 
         cls.creation_date = strftime("%Y-%m-%d %H:%M:%S", gmtime())
         cls.modified_date = strftime("%Y-%m-%d %H:%M:%S", gmtime())
-        io_controlnetwork.to_isis('test.net', dfs, mode='wb', targetname='Moon')
+        io_controlnetwork.to_isis('test.net', df, cls.serials, mode='wb', targetname='Moon')
 
         cls.header_message_size = 78
         cls.point_start_byte = 65614 # 66949
 
     def test_create_buffer_header(self):
-        self.npts = 5
+        npts = 5
         serial_times = {295: '1971-07-31T01:24:11.754',
                         296: '1971-07-31T01:24:36.970'}
-        self.serials = ['APOLLO15/METRIC/{}'.format(i) for i in serial_times.values()]
-        columns = ['point_id', 'point_type', 'serialnumber', 'measure_type', 'x', 'y', 'node_id']
+        serials = {i:'APOLLO15/METRIC/{}'.format(j) for i, j in enumerate(serial_times.values())}
+        columns = ['point_id', 'point_type', 'serialnumber', 'measure_type', 'x', 'y', 'image_index']
 
         data = []
         for i in range(self.npts):
-            data.append((i, 2, self.serials[0], 2, 0, 0, 0))
-            data.append((i, 2, self.serials[1], 2, 0, 0, 1))
+            data.append((i, 2, serials[0], 2, 0, 0, 0))
+            data.append((i, 2, serials[1], 2, 0, 0, 1))
 
-        dfs = [pd.DataFrame(data, columns=columns)]
+        df = pd.DataFrame(data, columns=columns)
 
         self.creation_date = strftime("%Y-%m-%d %H:%M:%S", gmtime())
         self.modified_date = strftime("%Y-%m-%d %H:%M:%S", gmtime())
-        io_controlnetwork.to_isis('test.net', dfs, mode='wb', targetname='Moon')
+        io_controlnetwork.to_isis('test.net', df, serials, mode='wb', targetname='Moon')
 
         self.header_message_size = 78
         self.point_start_byte = 65614 # 66949
@@ -86,7 +88,7 @@ class TestWriteIsisControlNetwork(unittest.TestCase):
                 self.assertEqual(str(i), point_protocol.id)
                 self.assertEqual(2, point_protocol.type)
                 for m in point_protocol.measures:
-                    self.assertTrue(m.serialnumber in self.serials)
+                    self.assertTrue(m.serialnumber in self.serials.values())
                     self.assertEqual(2, m.type)
 
     def test_create_pvl_header(self):
-- 
GitLab