From c3854bdb1f0739f579aa761f5a77b567e581e985 Mon Sep 17 00:00:00 2001
From: Joakim Soderlund <joakim.soderlund@gmail.com>
Date: Sun, 27 Aug 2023 10:42:43 +0200
Subject: [PATCH] Adjust tests for new type comparison lint

---
 tests/test_flavors.py | 4 ++--
 tests/test_stories.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/test_flavors.py b/tests/test_flavors.py
index 711a7fc..fe212b3 100644
--- a/tests/test_flavors.py
+++ b/tests/test_flavors.py
@@ -40,8 +40,8 @@ class TestFlavor:
         a = flavor.A
         b = flavor.B
 
-        assert type(a) == flavor
-        assert type(b) == flavor
+        assert type(a) is flavor
+        assert type(b) is flavor
         assert a.value == 1
         assert b.value == 2
         assert a is flavor.A
diff --git a/tests/test_stories.py b/tests/test_stories.py
index 78af850..d087b72 100644
--- a/tests/test_stories.py
+++ b/tests/test_stories.py
@@ -210,7 +210,7 @@ class TestStory:
 
         assert story.flavors is not flavors
         assert story.flavors == {flavor.A}
-        assert type(story.flavors) == set
+        assert type(story.flavors) is set
 
     def test_merge_without_parameters(self, story):
         """