|
3 | 3 | import os
|
4 | 4 | from test import test_support
|
5 | 5 |
|
| 6 | +def isinf(x): |
| 7 | + return x * 0.5 == x |
| 8 | + |
| 9 | +def isnan(x): |
| 10 | + return x != x |
| 11 | + |
6 | 12 | class FormatFunctionsTestCase(unittest.TestCase):
|
7 | 13 |
|
8 | 14 | def setUp(self):
|
@@ -128,13 +134,78 @@ def test_repr(self):
|
128 | 134 | self.assertEqual(v, eval(repr(v)))
|
129 | 135 | floats_file.close()
|
130 | 136 |
|
| 137 | +# Beginning with Python 2.6 float has cross platform compatible |
| 138 | +# ways to create and representate inf and nan |
| 139 | +class InfNanTest(unittest.TestCase): |
| 140 | + def test_inf_from_str(self): |
| 141 | + self.assert_(isinf(float("inf"))) |
| 142 | + self.assert_(isinf(float("+inf"))) |
| 143 | + self.assert_(isinf(float("-inf"))) |
| 144 | + |
| 145 | + self.assertEqual(repr(float("inf")), "inf") |
| 146 | + self.assertEqual(repr(float("+inf")), "inf") |
| 147 | + self.assertEqual(repr(float("-inf")), "-inf") |
| 148 | + |
| 149 | + self.assertEqual(repr(float("INF")), "inf") |
| 150 | + self.assertEqual(repr(float("+Inf")), "inf") |
| 151 | + self.assertEqual(repr(float("-iNF")), "-inf") |
| 152 | + |
| 153 | + self.assertEqual(str(float("inf")), "inf") |
| 154 | + self.assertEqual(str(float("+inf")), "inf") |
| 155 | + self.assertEqual(str(float("-inf")), "-inf") |
| 156 | + |
| 157 | + self.assertRaises(ValueError, float, "info") |
| 158 | + self.assertRaises(ValueError, float, "+info") |
| 159 | + self.assertRaises(ValueError, float, "-info") |
| 160 | + self.assertRaises(ValueError, float, "in") |
| 161 | + self.assertRaises(ValueError, float, "+in") |
| 162 | + self.assertRaises(ValueError, float, "-in") |
| 163 | + |
| 164 | + def test_inf_as_str(self): |
| 165 | + self.assertEqual(repr(1e300 * 1e300), "inf") |
| 166 | + self.assertEqual(repr(-1e300 * 1e300), "-inf") |
| 167 | + |
| 168 | + self.assertEqual(str(1e300 * 1e300), "inf") |
| 169 | + self.assertEqual(str(-1e300 * 1e300), "-inf") |
| 170 | + |
| 171 | + def test_nan_from_str(self): |
| 172 | + self.assert_(isnan(float("nan"))) |
| 173 | + self.assert_(isnan(float("+nan"))) |
| 174 | + self.assert_(isnan(float("-nan"))) |
| 175 | + |
| 176 | + self.assertEqual(repr(float("nan")), "nan") |
| 177 | + self.assertEqual(repr(float("+nan")), "nan") |
| 178 | + self.assertEqual(repr(float("-nan")), "nan") |
| 179 | + |
| 180 | + self.assertEqual(repr(float("NAN")), "nan") |
| 181 | + self.assertEqual(repr(float("+NAn")), "nan") |
| 182 | + self.assertEqual(repr(float("-NaN")), "nan") |
| 183 | + |
| 184 | + self.assertEqual(str(float("nan")), "nan") |
| 185 | + self.assertEqual(str(float("+nan")), "nan") |
| 186 | + self.assertEqual(str(float("-nan")), "nan") |
| 187 | + |
| 188 | + self.assertRaises(ValueError, float, "nana") |
| 189 | + self.assertRaises(ValueError, float, "+nana") |
| 190 | + self.assertRaises(ValueError, float, "-nana") |
| 191 | + self.assertRaises(ValueError, float, "na") |
| 192 | + self.assertRaises(ValueError, float, "+na") |
| 193 | + self.assertRaises(ValueError, float, "-na") |
| 194 | + |
| 195 | + def test_nan_as_str(self): |
| 196 | + self.assertEqual(repr(1e300 * 1e300 * 0), "nan") |
| 197 | + self.assertEqual(repr(-1e300 * 1e300 * 0), "nan") |
| 198 | + |
| 199 | + self.assertEqual(str(1e300 * 1e300 * 0), "nan") |
| 200 | + self.assertEqual(str(-1e300 * 1e300 * 0), "nan") |
131 | 201 |
|
132 | 202 | def test_main():
|
133 | 203 | test_support.run_unittest(
|
134 | 204 | FormatFunctionsTestCase,
|
135 | 205 | UnknownFormatTestCase,
|
136 | 206 | IEEEFormatTestCase,
|
137 |
| - #ReprTestCase |
| 207 | + ReprTestCase, |
| 208 | + InfNanTest, |
138 | 209 | )
|
139 | 210 |
|
140 | 211 | if __name__ == '__main__':
|
|
0 commit comments