import unittest import pygame from display import Display class TestDisplay(unittest.TestCase): def test_can_create(self): d = Display() self.assertNotEqual(d, None) def test_can_update(self): d = Display() surf = pygame.Surface((300,300)) sprite = pygame.sprite.Sprite() sprite.image = surf sprite.rect = pygame.Rect((0,0),(300,300)) group = pygame.sprite.Group() group.add(sprite) d.update([group])