python中py文件中不能传参到基础类调用基础类方法

问题遇到的现象和发生背景

照着别人敲的代码运行失败：
做一个简单的关键字驱动测试架构报错，想要在case.py文件传参数进基础类KeyDemo中报错
TypeError: KeyDemo() takes no arguments

问题相关代码，请勿粘贴截图

基础类

from selenium import webdriver


# 创建指定的浏览器对象
def browser(type_):
    try:
        return getattr(webdriver, type_)()
    except:
        return webdriver.Chrome()


class KeyDemo:

    # 构造函数
    def __init__(self, type_):
        self.driver = browser(type_)

    def open(self, url):
        self.driver.get(url)

case.py

from key_demo.key_worddemo import KeyDemo

wi = KeyDemo('Chrome')
wi.open("http://www.baidu.com")

运行结果及报错内容

Traceback (most recent call last):
  File "C:\Users\mihxy\PycharmProjects\key_demo\cases\case.py", line 7, in <module>
    wi = KeyDemo('Chrome')
TypeError: KeyDemo() takes no arguments

Process finished with exit code 1

我的解答思路和尝试过的方法

修改过包名和类名，确认不敏感

检查过init写法，是--

有查到一种说法：
类名(object)可以
类名（）也可以
类名（参数）不可以，不能够传参数

我想要达到的结果

想问一下，以我上面这种方式直接通过类名()传参的方式是否有问题，如果没问题，应该怎么修改
能够达成同等效果的也可以，感谢

问题排查：
有一个问题大概率影响上面的问题
我点击我导入的KeyDemo进入了--init--.py

class KeyDemo:
    pass

而不是进入我写好的，我想要导的那个包
将--init--.py中代码删了，报错

Cannot find reference 'KeyDemo' in '__init__.py' 

报错信息：

Traceback (most recent call last):
  File "C:\Users\mihxy\PycharmProjects\key_demo\cases\case.py", line 5, in <module>
    from key_demo.key_worddemo import KeyDemo
ImportError: cannot import name 'KeyDemo' from 'key_demo.key_worddemo' (C:\Users\mihxy\PycharmProjects\key_demo\key_worddemo\__init__.py)

Process finished with exit code 1

点击KeyDemo会提示在--init--.py中新建KeyDemo，就是生成最开始的代码

包的级别：

1、key-demo     
      2、cases                     3、 case.py
       2、key-worddemo      3、key-demo.py    3、 --init--.py

如下面这样运行没有问题的，估计是导入模块引起的问题：

# 创建指定的浏览器对象
def browser(type_):
    try:
        return getattr(webdriver, type_)()
    except:
        return webdriver.Chrome()


class KeyDemo:

    # 构造函数
    def __init__(self, type_):
        self.driver = browser(type_)

    def open(self, url):
        self.driver.get(url)
        return self.driver


wi = KeyDemo('Chrome')
page=wi.open("http://www.baidu.com")
print(page.page_source)
page.close()

你的代码运行没错，你将代码放在同一个py试试

有一个问题大概率影响上面的问题
我点击我导入的KeyDemo进入了--init--.py

class KeyDemo:
    pass

而不是进入我写好的，我想要导的那个包
将--init--.py中代码删了，报错

Cannot find reference 'KeyDemo' in '__init__.py' 

报错信息：

Traceback (most recent call last):
  File "C:\Users\mihxy\PycharmProjects\key_demo\cases\case.py", line 5, in <module>
    from key_demo.key_worddemo import KeyDemo
ImportError: cannot import name 'KeyDemo' from 'key_demo.key_worddemo' (C:\Users\mihxy\PycharmProjects\key_demo\key_worddemo\__init__.py)

Process finished with exit code 1

点击KeyDemo会提示在--init--.py中新建KeyDemo，就是生成最开始的代码

包的级别：
1、key_demo
2、cases 3、 case.py
2、key_worddemo 3、key_demo.py 3、 --init--.py