1  import unittest
 2  from blottertrax.description_provider import DescriptionProvider
 3  
 4  
 5  class MyTestCase(unittest.TestCase):
 6      def test_format_network_to_friendly_name(self):
 7  
 8          networks = [
 9              ('social network', 'https://facebook.com/linkedpage', '[facebook](https://facebook.com/linkedpage)'),
10              ('social network', 'https://twitter.com/linkedpage', '[twitter](https://twitter.com/linkedpage)'),
11              ('social network', 'https://instagram.com/linkedpage', '[instagram](https://instagram.com/linkedpage)'),
12              ('discogs', 'https://discogs.com/linkedpage', '[discogs](https://discogs.com/linkedpage)'),
13              (
14                  'some other database',
15                  'https://some-other-database.com/linkedpage',
16                  '[some other database](https://some-other-database.com/linkedpage)'
17              ),
18              (
19                  'wikipedia',
20                  'https://en.wikipedia.org/wiki/Caravan_(band)',
21                  r'[wikipedia](https://en.wikipedia.org/wiki/Caravan_\(band\))'
22              ),
23          ]
24          for network_type, target, expected in networks:
25              formatted = DescriptionProvider.format_network_to_friendly_name({
26                  'target': target,
27                  'type': network_type
28              })
29  
30              self.assertEqual(expected, formatted)