diff --git a/rock_paper_sand/multi_level_set_test.py b/rock_paper_sand/multi_level_set_test.py index dfbfe1c..ec6128a 100644 --- a/rock_paper_sand/multi_level_set_test.py +++ b/rock_paper_sand/multi_level_set_test.py @@ -198,7 +198,9 @@ def test_valid_set( is_member: bool, ) -> None: test_set = multi_level_set.MultiLevelSet.from_string(set_str) - self.assertEqual(is_member, number in test_set) + self.assertEqual( + is_member, multi_level_set.MultiLevelNumber(number) in test_set + ) @parameterized.parameters( ".",