def test_merge_attributes_test_with_str_dtype(self):
ne = NetworkEdge()
attrs = [
Attribute(name='x', value=['1'], data_type='list_of_string'),
Attribute(name='x', value='3')
]
res = ne._merge_attributes(attrs_list=attrs)
self.assertEqual('x', res.get_name())
self.assertEqual(2, len(res.get_value()))
self.assertTrue('1' in res.get_value())
self.assertTrue('3' in res.get_value())
self.assertEqual('list_of_string', res.get_data_type())
def test_merge_attributes_test_where_dtype_mixed_string_none(self):
ne = NetworkEdge()
attrs = [
Attribute(name='x', value='1', data_type='string'),
Attribute(name='x', value='2')
]
res = ne._merge_attributes(attrs_list=attrs)
self.assertEqual('x', res.get_name())
self.assertEqual(2, len(res.get_value()))
self.assertTrue('1' in res.get_value())
self.assertTrue('2' in res.get_value())
self.assertEqual('list_of_string', res.get_data_type())
def test_merge_attributes_test_with_integer_dtype(self):
ne = NetworkEdge()
attrs = [
Attribute(name='x', value=1, data_type='integer'),
Attribute(name='x', value=[2, 3], data_type='list_of_integer')
]
res = ne._merge_attributes(attrs_list=attrs)
self.assertEqual('x', res.get_name())
self.assertEqual(3, len(res.get_value()))
self.assertTrue(1 in res.get_value())
self.assertTrue(2 in res.get_value())
self.assertTrue(3 in res.get_value())
self.assertEqual('list_of_integer', res.get_data_type())