def test_devicecount(stc):
seq_root = ET.fromstring(stak_estimation)
port_tag_location_dict = {
"West Port Group": ["10.14.16.27/2/7", "10.14.16.17/2/7"],
"East Port Group": ["10.14.16.37/2/7", "10.14.16.47/2/8"],
}
est_util = SequencerEstimator(root=seq_root)
est_util.set_port_dict(port_tag_location_dict)
deviceCountEstimate = est_util.estimate_device_count()
assert deviceCountEstimate["10.14.16.27/2/7"] == 5
assert deviceCountEstimate["10.14.16.17/2/7"] == 5
assert deviceCountEstimate["10.14.16.37/2/7"] == 1
assert deviceCountEstimate["10.14.16.47/2/8"] == 1
def test_loadtemplatecommand_not_present(stc):
seq_root = ET.fromstring(sequencer_xml)
port_tag_location_dict = {
"West Port Group": ["10.14.16.27/2/7", "10.14.16.17/2/7"],
"East Port Group": ["10.14.16.37/2/7", "10.14.16.47/2/8"],
}
est_util = SequencerEstimator(root=seq_root)
est_util.set_port_dict(port_tag_location_dict)
deviceCountEstimate = est_util.estimate_device_count()
assert deviceCountEstimate["10.14.16.27/2/7"] == 0
assert deviceCountEstimate["10.14.16.17/2/7"] == 0
assert deviceCountEstimate["10.14.16.37/2/7"] == 0
assert deviceCountEstimate["10.14.16.47/2/8"] == 0
