diff --git a/bifrost/tests/unit/test_inventory.py b/bifrost/tests/unit/test_inventory.py index 4c024d262..a895afab9 100644 --- a/bifrost/tests/unit/test_inventory.py +++ b/bifrost/tests/unit/test_inventory.py @@ -37,6 +37,6 @@ class TestBifrostInventoryUnit(base.TestCase): def test__val_or_none(self): array = ['no', '', 'yes'] self.assertEqual('no', inventory._val_or_none(array, 0)) - self.assertEqual(None, inventory._val_or_none(array, 1)) + self.assertIsNone(inventory._val_or_none(array, 1)) self.assertEqual('yes', inventory._val_or_none(array, 2)) - self.assertEqual(None, inventory._val_or_none(array, 4)) + self.assertIsNone(inventory._val_or_none(array, 4))