Merge pull request #72 from yassun7010/fix_row_type_check

fix: row_type check.
yassun7010 · Feb 8, 2024 · bc077e7 · bc077e7
2 parents b21fcdd + f80cb26
commit bc077e7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/turu-core/src/turu/core/mock/store.py b/turu-core/src/turu/core/mock/store.py
@@ -45,7 +45,10 @@ def provide_response(
 
  _row_type, _response = self._data.pop(0)
 
- if _row_type is not row_type:
+ if _row_type is not row_type and not (
+ row_type.__module__ == _row_type.__module__
+ and row_type.__name__ == _row_type.__name__
+ ):
  raise TuruMockResponseTypeMismatchError(row_type, _row_type, self._counter)
 
  if isinstance(_response, Exception):