示例#1
0
文件: tests.py 项目: mikoim/funstuff
class FieldTests(unittest.TestCase):
    def setUp(self):
        self.__x = random.randint(2, 128)
        self.__y = random.randint(2, 128)
        self.__field = Field(self.__y, self.__x)

    def test_zero(self):
        self.assertFalse(Field(0, 0).is_correct(0, 0))

    def test_size(self):
        self.assertEqual(self.__field.x, self.__x)
        self.assertEqual(self.__field.y, self.__y)

    def test_basic(self):
        n = random.random()
        x = random.randint(0, self.__x - 1)
        y = random.randint(0, self.__y - 1)
        self.__field[y][x] = n

        self.assertEqual(self.__field[y][x], n)

    def test_border(self):
        self.__field[0][0] = 0
        self.__field[self.__y - 1][self.__x - 1] = 0

    def test_method_is_correct(self):
        x = random.randint(0, self.__x - 1)
        y = random.randint(0, self.__y - 1)

        self.assertTrue(self.__field.is_correct(0, 0))
        self.assertTrue(self.__field.is_correct(y, x))
        self.assertTrue(not self.__field.is_correct(-1, 0))
        self.assertTrue(not self.__field.is_correct(0, -1))
        self.assertTrue(not self.__field.is_correct(self.__y, 0))
        self.assertTrue(not self.__field.is_correct(0, self.__x))
示例#2
0
    def __init__(self, field: Field, x: int, y: int):
        self._neighbor = {'grass': 0, 'herbivore': 0, 'carnivore': 0}

        if not field.is_correct(y, x):
            raise IndexError

        for dx, dy in [(0, 0), (0, -1), (1, -1), (1, 0), (1, 1), (0, 1), (-1, 1), (-1, 0), (-1, -1)]:
            cx = x + dx
            cy = y + dy

            if not field.is_correct(cy, cx):
                continue

            self._neighbor['grass'] += field[cy][cx].get_grass()
            self._neighbor['herbivore'] += field[cy][cx].get_herbivore()
            self._neighbor['carnivore'] += field[cy][cx].get_carnivore()
示例#3
0
    def __init__(self, field: Field, x: int, y: int):
        self.__neighbor = {}

        if not field.is_correct(y, x):
            raise IndexError

        for dx, dy in [(0, -1), (1, -1), (1, 0), (1, 1), (0, 1), (-1, 1), (-1, 0), (-1, -1)]:
            cx = x + dx
            cy = y + dy

            if not field.is_correct(cy, cx):
                continue

            class_name = type(field[cy][cx]).__name__

            if class_name in self.__neighbor:
                self.__neighbor[class_name] += 1
            else:
                self.__neighbor[class_name] = 1