def test_connect(self): sqlite3_mock = MagicMock() row_factory_mock = PropertyMock(return_value=None) type(sqlite3_mock).row_factory = row_factory_mock sqlite3.connect = sqlite3_mock db = Database('my_db') db.connect() sqlite3.connect.assert_called_with('my_db') self.assertIsInstance(db.conn, MagicMock)