def test_many_edges(self): tweet = Tweet('Hello #One #two #THREE', 'Thu Oct 30 18:10:49 +0000 2015') expected_edges = ['#one-#two', '#one-#three', '#three-#two'] edges = tweet.get_edges() self.assertEqual(len(expected_edges), len(edges)) for edge in edges: self.assertIn(str(edge), expected_edges)
def test_edges_for_no_hashtags(self): tweet = Tweet('Hello No hash', 'Thu Oct 30 18:10:49 +0000 2015') expected_edges = [] self.assertEqual(expected_edges, tweet.get_edges())
def test_one_edge(self): tweet = Tweet('Hello #Abra #Kadabra', 'Thu Oct 30 18:10:49 +0000 2015') expected_edges = ['#abra-#kadabra'] edges = tweet.get_edges() self.assertEqual(len(expected_edges), len(edges)) self.assertEqual(expected_edges[0], str(edges[0]))