53 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			53 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			Python
		
	
	
	
| from . import MuseDashTestBase
 | |
| 
 | |
| 
 | |
| class TestPlandoSettings(MuseDashTestBase):
 | |
|     options = {
 | |
|         "additional_song_count": 15,
 | |
|         "dlc_packs": {"Muse Plus"},
 | |
|         "include_songs": [
 | |
|             "Lunatic",
 | |
|             "Out of Sense",
 | |
|             "Magic Knight Girl",
 | |
|         ]
 | |
|     }
 | |
| 
 | |
|     def test_included_songs_didnt_grow_item_count(self) -> None:
 | |
|         muse_dash_world = self.get_world()
 | |
|         self.assertEqual(len(muse_dash_world.included_songs), 15, "Logical songs size grew when it shouldn't.")
 | |
| 
 | |
|     def test_included_songs_plando(self) -> None:
 | |
|         muse_dash_world = self.get_world()
 | |
|         songs = muse_dash_world.included_songs.copy()
 | |
|         songs.append(muse_dash_world.victory_song_name)
 | |
| 
 | |
|         self.assertIn("Lunatic", songs, "Logical songs is missing a plando song: Lunatic")
 | |
|         self.assertIn("Out of Sense", songs, "Logical songs is missing a plando song: Out of Sense")
 | |
|         self.assertIn("Magic Knight Girl", songs, "Logical songs is missing a plando song: Magic Knight Girl")
 | |
| 
 | |
| 
 | |
| class TestFilteredPlandoSettings(MuseDashTestBase):
 | |
|     options = {
 | |
|         "additional_song_count": 15,
 | |
|         "dlc_packs": {"MSR Anthology"},
 | |
|         "include_songs": [
 | |
|             "Operation Blade",
 | |
|             "Autumn Moods",
 | |
|             "Fireflies",
 | |
|         ]
 | |
|     }
 | |
| 
 | |
|     def test_included_songs_didnt_grow_item_count(self) -> None:
 | |
|         muse_dash_world = self.get_world()
 | |
|         self.assertEqual(len(muse_dash_world.included_songs), 15, "Logical songs size grew when it shouldn't.")
 | |
| 
 | |
|     # Tests for excluding included songs when the right dlc isn't enabled
 | |
|     def test_filtered_included_songs_plando(self) -> None:
 | |
|         muse_dash_world = self.get_world()
 | |
|         songs = muse_dash_world.included_songs.copy()
 | |
|         songs.append(muse_dash_world.victory_song_name)
 | |
| 
 | |
|         self.assertIn("Operation Blade", songs, "Logical songs is missing a plando song: Operation Blade")
 | |
|         self.assertIn("Autumn Moods", songs, "Logical songs is missing a plando song: Autumn Moods")
 | |
|         self.assertNotIn("Fireflies", songs, "Logical songs has added a filtered a plando song: Fireflies")
 |