Source code for game.test_suite.tests.test_initialization

import unittest

from game.common.enums import ObjectType
from game.common.avatar import Avatar
from game.common.items.item import Item
from game.common.stations.station import Station
from game.common.stations.occupiable_station import OccupiableStation
from game.common.map.tile import Tile
from game.common.map.wall import Wall
from game.utils.vector import Vector


class TestInitialization(unittest.TestCase):
    """
    `Test Avatar Notes:`

        This class tests the instantiation of different Objects within the project. Recall that every new class created
        needs a respective ObjectType enum created for it.
    """

    def setUp(self) -> None:
        self.item: Item = Item(10, 100)
        self.avatar: Avatar = Avatar(None)
        self.station: Station = Station(None)
        self.occupiable_station: OccupiableStation = OccupiableStation(None, None)
        self.tile: Tile = Tile(None)
        self.wall: Wall = Wall()
        self.vector: Vector = Vector()

    # tests that all objects have the correct ObjectType
    def test_object_init(self):
        self.assertEqual(self.item.object_type, ObjectType.ITEM)
        self.assertEqual(self.avatar.object_type, ObjectType.AVATAR)
        self.assertEqual(self.station.object_type, ObjectType.STATION)
        self.assertEqual(self.occupiable_station.object_type, ObjectType.OCCUPIABLE_STATION)
        self.assertEqual(self.tile.object_type, ObjectType.TILE)
        self.assertEqual(self.wall.object_type, ObjectType.WALL)
        self.assertEqual(self.vector.object_type, ObjectType.VECTOR)