48 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			48 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Python
		
	
	
	
from . import MM2TestBase
 | 
						|
from ..locations import (quick_man_locations, heat_man_locations, wily_1_locations, wily_2_locations,
 | 
						|
                         wily_3_locations, wily_4_locations, wily_5_locations, wily_6_locations,
 | 
						|
                         energy_pickups, etank_1ups)
 | 
						|
from ..names import *
 | 
						|
 | 
						|
 | 
						|
class TestAccess(MM2TestBase):
 | 
						|
    options = {
 | 
						|
        "consumables": "all"
 | 
						|
    }
 | 
						|
 | 
						|
    def test_time_stopper(self) -> None:
 | 
						|
        """Optional based on Enable Lasers setting, confirm these are the locations affected"""
 | 
						|
        locations = [*quick_man_locations, *energy_pickups["Quick Man Stage"], *etank_1ups["Quick Man Stage"]]
 | 
						|
        items = [["Time Stopper"]]
 | 
						|
        self.assertAccessDependency(locations, items)
 | 
						|
 | 
						|
    def test_item_2(self) -> None:
 | 
						|
        """Optional based on Yoku Block setting, confirm these are the locations affected"""
 | 
						|
        locations = [*heat_man_locations, *etank_1ups["Heat Man Stage"]]
 | 
						|
        items = [["Item 2 - Rocket"]]
 | 
						|
        self.assertAccessDependency(locations, items, True)
 | 
						|
 | 
						|
    def test_any_item(self) -> None:
 | 
						|
        locations = [flash_man_c2, quick_man_c1, crash_man_c3]
 | 
						|
        items = [["Item 1 - Propeller"], ["Item 2 - Rocket"], ["Item 3 - Bouncy"]]
 | 
						|
        self.assertAccessDependency(locations, items, True)
 | 
						|
        locations = [metal_man_c2, metal_man_c3]
 | 
						|
        items = [["Item 1 - Propeller"], ["Item 2 - Rocket"]]
 | 
						|
        self.assertAccessDependency(locations, items, True)
 | 
						|
 | 
						|
    def test_all_items(self) -> None:
 | 
						|
        locations = [flash_man_c2, quick_man_c1, crash_man_c3, metal_man_c2, metal_man_c3, *heat_man_locations,
 | 
						|
                     *etank_1ups["Heat Man Stage"], *wily_1_locations, *wily_2_locations, *wily_3_locations,
 | 
						|
                     *wily_4_locations, *wily_5_locations, *wily_6_locations, *etank_1ups["Wily Stage 1"],
 | 
						|
                     *etank_1ups["Wily Stage 2"], *etank_1ups["Wily Stage 3"], *etank_1ups["Wily Stage 4"],
 | 
						|
                     *energy_pickups["Wily Stage 1"], *energy_pickups["Wily Stage 2"], *energy_pickups["Wily Stage 3"],
 | 
						|
                     *energy_pickups["Wily Stage 4"]]
 | 
						|
        items = [["Item 1 - Propeller", "Item 2 - Rocket", "Item 3 - Bouncy"]]
 | 
						|
        self.assertAccessDependency(locations, items)
 | 
						|
 | 
						|
    def test_crash_bomber(self) -> None:
 | 
						|
        locations = [flash_man_c3, flash_man_c4, wily_2_c5, wily_2_c6, wily_3_c1, wily_3_c2,
 | 
						|
                     wily_4, wily_stage_4]
 | 
						|
        items = [["Crash Bomber"]]
 | 
						|
        self.assertAccessDependency(locations, items)
 |