forked from python/cpython
-
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'upstream/main'
- Loading branch information
Showing
19 changed files
with
702 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,146 @@ | ||
| import unittest | ||
| import warnings | ||
|
|
||
| from test.test_capi.test_getargs import (BadComplex, BadComplex2, Complex, | ||
| FloatSubclass, Float, BadFloat, | ||
| BadFloat2, ComplexSubclass) | ||
| from test.support import import_helper | ||
|
|
||
|
|
||
| _testcapi = import_helper.import_module('_testcapi') | ||
|
|
||
| NULL = None | ||
|
|
||
| class BadComplex3: | ||
| def __complex__(self): | ||
| raise RuntimeError | ||
|
|
||
|
|
||
| class CAPIComplexTest(unittest.TestCase): | ||
| def test_check(self): | ||
| # Test PyComplex_Check() | ||
| check = _testcapi.complex_check | ||
|
|
||
| self.assertTrue(check(1+2j)) | ||
| self.assertTrue(check(ComplexSubclass(1+2j))) | ||
| self.assertFalse(check(Complex())) | ||
| self.assertFalse(check(3)) | ||
| self.assertFalse(check(3.0)) | ||
| self.assertFalse(check(object())) | ||
|
|
||
| # CRASHES check(NULL) | ||
|
|
||
| def test_checkexact(self): | ||
| # PyComplex_CheckExact() | ||
| checkexact = _testcapi.complex_checkexact | ||
|
|
||
| self.assertTrue(checkexact(1+2j)) | ||
| self.assertFalse(checkexact(ComplexSubclass(1+2j))) | ||
| self.assertFalse(checkexact(Complex())) | ||
| self.assertFalse(checkexact(3)) | ||
| self.assertFalse(checkexact(3.0)) | ||
| self.assertFalse(checkexact(object())) | ||
|
|
||
| # CRASHES checkexact(NULL) | ||
|
|
||
| def test_fromccomplex(self): | ||
| # Test PyComplex_FromCComplex() | ||
| fromccomplex = _testcapi.complex_fromccomplex | ||
|
|
||
| self.assertEqual(fromccomplex(1+2j), 1.0+2.0j) | ||
|
|
||
| def test_fromdoubles(self): | ||
| # Test PyComplex_FromDoubles() | ||
| fromdoubles = _testcapi.complex_fromdoubles | ||
|
|
||
| self.assertEqual(fromdoubles(1.0, 2.0), 1.0+2.0j) | ||
|
|
||
| def test_realasdouble(self): | ||
| # Test PyComplex_RealAsDouble() | ||
| realasdouble = _testcapi.complex_realasdouble | ||
|
|
||
| self.assertEqual(realasdouble(1+2j), 1.0) | ||
| self.assertEqual(realasdouble(-1+0j), -1.0) | ||
| self.assertEqual(realasdouble(4.25), 4.25) | ||
| self.assertEqual(realasdouble(-1.0), -1.0) | ||
| self.assertEqual(realasdouble(42), 42.) | ||
| self.assertEqual(realasdouble(-1), -1.0) | ||
|
|
||
| # Test subclasses of complex/float | ||
| self.assertEqual(realasdouble(ComplexSubclass(1+2j)), 1.0) | ||
| self.assertEqual(realasdouble(FloatSubclass(4.25)), 4.25) | ||
|
|
||
| # Test types with __complex__ dunder method | ||
| # Function doesn't support classes with __complex__ dunder, see #109598 | ||
| self.assertRaises(TypeError, realasdouble, Complex()) | ||
|
|
||
| # Test types with __float__ dunder method | ||
| self.assertEqual(realasdouble(Float()), 4.25) | ||
| self.assertRaises(TypeError, realasdouble, BadFloat()) | ||
| with self.assertWarns(DeprecationWarning): | ||
| self.assertEqual(realasdouble(BadFloat2()), 4.25) | ||
|
|
||
| self.assertRaises(TypeError, realasdouble, object()) | ||
|
|
||
| # CRASHES realasdouble(NULL) | ||
|
|
||
| def test_imagasdouble(self): | ||
| # Test PyComplex_ImagAsDouble() | ||
| imagasdouble = _testcapi.complex_imagasdouble | ||
|
|
||
| self.assertEqual(imagasdouble(1+2j), 2.0) | ||
| self.assertEqual(imagasdouble(1-1j), -1.0) | ||
| self.assertEqual(imagasdouble(4.25), 0.0) | ||
| self.assertEqual(imagasdouble(42), 0.0) | ||
|
|
||
| # Test subclasses of complex/float | ||
| self.assertEqual(imagasdouble(ComplexSubclass(1+2j)), 2.0) | ||
| self.assertEqual(imagasdouble(FloatSubclass(4.25)), 0.0) | ||
|
|
||
| # Test types with __complex__ dunder method | ||
| # Function doesn't support classes with __complex__ dunder, see #109598 | ||
| self.assertEqual(imagasdouble(Complex()), 0.0) | ||
|
|
||
| # Function returns 0.0 anyway, see #109598 | ||
| self.assertEqual(imagasdouble(object()), 0.0) | ||
|
|
||
| # CRASHES imagasdouble(NULL) | ||
|
|
||
| def test_asccomplex(self): | ||
| # Test PyComplex_AsCComplex() | ||
| asccomplex = _testcapi.complex_asccomplex | ||
|
|
||
| self.assertEqual(asccomplex(1+2j), 1.0+2.0j) | ||
| self.assertEqual(asccomplex(-1+2j), -1.0+2.0j) | ||
| self.assertEqual(asccomplex(4.25), 4.25+0.0j) | ||
| self.assertEqual(asccomplex(-1.0), -1.0+0.0j) | ||
| self.assertEqual(asccomplex(42), 42+0j) | ||
| self.assertEqual(asccomplex(-1), -1.0+0.0j) | ||
|
|
||
| # Test subclasses of complex/float | ||
| self.assertEqual(asccomplex(ComplexSubclass(1+2j)), 1.0+2.0j) | ||
| self.assertEqual(asccomplex(FloatSubclass(4.25)), 4.25+0.0j) | ||
|
|
||
| # Test types with __complex__ dunder method | ||
| self.assertEqual(asccomplex(Complex()), 4.25+0.5j) | ||
| self.assertRaises(TypeError, asccomplex, BadComplex()) | ||
| with self.assertWarns(DeprecationWarning): | ||
| self.assertEqual(asccomplex(BadComplex2()), 4.25+0.5j) | ||
| with warnings.catch_warnings(): | ||
| warnings.simplefilter("error", DeprecationWarning) | ||
| self.assertRaises(DeprecationWarning, asccomplex, BadComplex2()) | ||
| self.assertRaises(RuntimeError, asccomplex, BadComplex3()) | ||
|
|
||
| # Test types with __float__ dunder method | ||
| self.assertEqual(asccomplex(Float()), 4.25+0.0j) | ||
| self.assertRaises(TypeError, asccomplex, BadFloat()) | ||
| with self.assertWarns(DeprecationWarning): | ||
| self.assertEqual(asccomplex(BadFloat2()), 4.25+0.0j) | ||
|
|
||
| self.assertRaises(TypeError, asccomplex, object()) | ||
|
|
||
| # CRASHES asccomplex(NULL) | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| unittest.main() |
Oops, something went wrong.