update tests

Author: scetron

tests/integration/test_parsing_igmp.py

@@ -7,25 +7,24 @@
 
 
 def validate_igmp(df: pd.DataFrame):
-    '''Validate the routes table for all values'''
+    """Validate the routes table for all values"""
 
-    assert (df.vrf != '').all()
-    assert (df.group != '').all()
+    assert (df.vrf != "").all()
+    assert (df.group != "").all()
 
 
-
-@ pytest.mark.parsing
-@ pytest.mark.igmp
-@ pytest.mark.parametrize('table', ['igmp'])
-@ pytest.mark.parametrize('datadir', DATADIR)
+@pytest.mark.parsing
+@pytest.mark.igmp
+@pytest.mark.parametrize("table", ["igmp"])
+@pytest.mark.parametrize("datadir", DATADIR)
 # pylint: disable=unused-argument
 def test_igmp_parsing(table, datadir, get_table_data):
-    '''Main workhorse routine to test parsed output for Routes'''
+    """Main workhorse routine to test parsed output for igmp"""
 
     df = get_table_data
 
     ns_dict = {
-        'eos': 2,
+        "multicast": 2,
     }
 
     assert not df.empty

tests/integration/test_parsing_mroutes.py

@@ -7,26 +7,25 @@
 
 
 def validate_mroutes(df: pd.DataFrame):
-    '''Validate the routes table for all values'''
+    """Validate the routes table for all values"""
 
     # assert (df.vrf != '').all()
-    assert (df.group != '').all()
-    assert (df.source != '').all()
 
+    assert (df.group != "").all()
+    assert (df.source != "").all()
 
-@ pytest.mark.parsing
-@ pytest.mark.mroutes
-@ pytest.mark.parametrize('table', ['mroutes'])
-@ pytest.mark.parametrize('datadir', DATADIR)
+
+@pytest.mark.parsing
+@pytest.mark.mroutes
+@pytest.mark.parametrize("table", ["mroutes"])
+@pytest.mark.parametrize("datadir", DATADIR)
 # pylint: disable=unused-argument
 def test_mroutes_parsing(table, datadir, get_table_data):
-    '''Main workhorse routine to test parsed output for Routes'''
+    """Main workhorse routine to test parsed output for mRoutes"""
 
     df = get_table_data
 
-    ns_dict = {
-        'eos': 2
-    }
+    ns_dict = {"multicast": 6}
 
     assert not df.empty
     validate_host_shape(df, ns_dict)