Fix incorrect use of NewType-defined type

dseomn · Sep 24, 2023 · 005d195 · 005d195
1 parent 800629c
commit 005d195
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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(
         ".",