Generalize card image test

zedeus · May 2, 2020 · 5d561ea · 5d561ea
1 parent 9197c68
commit 5d561ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_card.py b/tests/test_card.py
@@ -107,7 +107,7 @@ def test_card(self, tweet, title, description, destination, large):
  card = Card(Conversation.main + " ")
  self.assert_text(title, card.title)
  self.assert_text(destination, card.destination)
- self.assertIn('_img', self.get_image_url(card.image + ' img'))
+ self.assertIn('twimg', self.get_image_url(card.image + ' img'))
  if len(description) > 0:
  self.assert_text(description, card.description)
  if large:
@@ -130,7 +130,7 @@ def test_card_playable(self, tweet, title, description, destination):
  card = Card(Conversation.main + " ")
  self.assert_text(title, card.title)
  self.assert_text(destination, card.destination)
- self.assertIn('_img', self.get_image_url(card.image + ' img'))
+ self.assertIn('twimg', self.get_image_url(card.image + ' img'))
  self.assert_element_visible('.card-overlay')
  if len(description) > 0:
  self.assert_text(description, card.description)