Skip to content

Commit bc077e7

Browse files
authored
Merge pull request #72 from yassun7010/fix_row_type_check
fix: row_type check.
2 parents b21fcdd + f80cb26 commit bc077e7

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

turu-core/src/turu/core/mock/store.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,10 @@ def provide_response(
4545

4646
_row_type, _response = self._data.pop(0)
4747

48-
if _row_type is not row_type:
48+
if _row_type is not row_type and not (
49+
row_type.__module__ == _row_type.__module__
50+
and row_type.__name__ == _row_type.__name__
51+
):
4952
raise TuruMockResponseTypeMismatchError(row_type, _row_type, self._counter)
5053

5154
if isinstance(_response, Exception):

0 commit comments

Comments
 (0)