test_terraindisplay.py 文件源码

def setUp(self):
        self.ter1 = Terrain(100, 100)   # all black
        self.ter2 = Terrain(200, 200)   # all white
        for x in range(self.ter2.width):
            for y in range(self.ter2.length):
                self.ter2[x, y] = 1
        self.ter3 = Terrain(100, 100)   # main diagonal is increasing brightness downwards
        for x in range(self.ter3.width):
            for y in range(self.ter3.length):
                if x == y:
                    self.ter3[x, y] = float(y) / self.ter3.length
        self.ter4 = Terrain(200, 100)   # checkerboard pattern
        for x in range(self.ter4.width):
            for y in range(self.ter4.length):
                self.ter4[x, y] = 1 if (x + y) % 2 == 0 else 0