Skip to content

Commit f80cb26

Browse files
committed
fix: row_type check.
1 parent b21fcdd commit f80cb26

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)