Update unittest for graph with sections

VIDA-NYU · Aug 16, 2017 · 349d5c7 · 349d5c7
1 parent 347dad2
commit 349d5c7
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/tests/test_graph.py b/tests/test_graph.py
@@ -113,12 +113,14 @@ def setUpClass(cls):
   files:
   - "/usr/lib/2_one.so"
   - "/etc/2_two.cfg"
+  meta: {"section": "libs"}
 - name: python
   version: "2.7"
   size: 5000000
   packfiles: true
   files:
   - "/usr/bin/python"
+  meta: {"section": "python"}
 - name: unused
   version: "0.1"
   size: 100
@@ -258,12 +260,15 @@ def test_simple(self):
 }
 """,
             {'packages': [{'name': 'pkg1', 'version': '1.0',
-                           'files': ['/usr/bin/wc']},
+                           'files': ['/usr/bin/wc'],
+                           'section': None},
                           {'name': 'pkg2', 'version': '1.0',
                            'files': ['/etc/2_two.cfg',
-                                     '/usr/lib/2_one.so']},
+                                     '/usr/lib/2_one.so'],
+                           'section': 'libs'},
                           {'name': 'python', 'version': '2.7',
-                           'files': ['/usr/bin/python']}],
+                           'files': ['/usr/bin/python'],
+                           'section': 'python'}],
              'other_files': ['/bin/sh',
                              '/some/dir/drive.py',
                              '/some/dir/experiment',