diff --git a/dictorm/test/test_dictorm.py b/dictorm/test/test_dictorm.py index 6253123..06bbf72 100755 --- a/dictorm/test/test_dictorm.py +++ b/dictorm/test/test_dictorm.py @@ -53,11 +53,6 @@ def assertRaisesAny(cls, exps, func, a=None, kw=None): if isinstance(e, exps): return raise Exception('Did not raise one of the exceptions provided!') - @classmethod - def assertType(cls, a, b): - if not isinstance(a, b): - raise TypeError('{0} is not type {1}'.format(str(a), b0)) - def assertEqualNoRefs(self, a, b): return self.assertEqual(_no_refs(a), _no_refs(b)) @@ -878,11 +873,11 @@ def test_results_cache(self): subordinates = bob['subordinates'] for sub in subordinates: - self.assertType(sub, dictorm.Dict) + assert isinstance(sub, dictorm.Dict) # Error would be raised if subordinates isn't cached bob._table.get_where = error for sub in subordinates: - self.assertType(sub, dictorm.Dict) + assert isinstance(sub, dictorm.Dict) def test_reference_order(self): """