From 835406da65cdcf3602b4af98c720bafaf4f1062e Mon Sep 17 00:00:00 2001
From: Roman Mogylatov <rmogilatov@gmail.com>
Date: Wed, 13 Oct 2021 19:04:05 -0400
Subject: [PATCH] Migrate misc wiring tests

---
 .../test_module_as_package_py36.py            | 20 +++++++++++++++++++
 tests/unit/wiring/test_wiring_py36.py         | 13 ------------
 2 files changed, 20 insertions(+), 13 deletions(-)
 create mode 100644 tests/unit/wiring/provider_ids/test_module_as_package_py36.py

diff --git a/tests/unit/wiring/provider_ids/test_module_as_package_py36.py b/tests/unit/wiring/provider_ids/test_module_as_package_py36.py
new file mode 100644
index 00000000..8ad74bbe
--- /dev/null
+++ b/tests/unit/wiring/provider_ids/test_module_as_package_py36.py
@@ -0,0 +1,20 @@
+"""Tests for wiring to module as package."""
+
+from pytest import fixture
+
+from wiringsamples import module
+from wiringsamples.service import Service
+from wiringsamples.container import Container
+
+
+@fixture
+def container():
+    container = Container()
+    yield container
+    container.unwire()
+
+
+def test_module_as_package_wiring(container: Container):
+    # See: https://github.com/ets-labs/python-dependency-injector/issues/481
+    container.wire(packages=[module])
+    assert isinstance(module.service, Service)
diff --git a/tests/unit/wiring/test_wiring_py36.py b/tests/unit/wiring/test_wiring_py36.py
index d6f4cf04..798d8ab9 100644
--- a/tests/unit/wiring/test_wiring_py36.py
+++ b/tests/unit/wiring/test_wiring_py36.py
@@ -4,7 +4,6 @@ from dependency_injector.wiring import (
     Provide,
     Closing,
 )
-from dependency_injector import containers, errors
 
 # Runtime import to avoid syntax errors in samples on Python < 3.5
 import os
@@ -29,18 +28,6 @@ from wiringsamples.service import Service
 from wiringsamples.container import Container
 
 
-class ModuleAsPackageTest(unittest.TestCase):
-
-    def setUp(self):
-        self.container = Container(config={"a": {"b": {"c": 10}}})
-        self.addCleanup(self.container.unwire)
-
-    def test_module_as_package_wiring(self):
-        # See: https://github.com/ets-labs/python-dependency-injector/issues/481
-        self.container.wire(packages=[module])
-        self.assertIsInstance(module.service, Service)
-
-
 class WiringAndQueue(unittest.TestCase):
 
     def test_wire_queue(self) -> None: