def test_negative_step(self):
with self.assertRaises(StopIteration):
# Should be empty so StopIteration should be raised
next(iter(float_range2(1, 6, -1)))
self.assertEqual(list(float_range2(5, 0, -1)), [5, 4, 3, 2, 1])
self.assertEqual(list(float_range2(0.5, 6)),
[0.5, 1.5, 2.5, 3.5, 4.5, 5.5])
self.assertEqual(list(float_range2(6, 1, -0.5)),
[6, 5.5, 5.0, 4.5, 4.0, 3.5, 3.0, 2.5, 2.0, 1.5])
def test_has_length(self):
self.assertEqual(len(float_range2(100)), 100)
self.assertEqual(len(float_range2(1, 100)), 99)
self.assertEqual(len(float_range2(1, 11, 2)), 5)
self.assertEqual(len(float_range2(0.5, 7, 0.75)), 9)
self.assertEqual(len(float_range2(1000000)), 1000000)
self.assertEqual(len(float_range2(11, 1.2, -2)), 5)
self.assertEqual(len(float_range2(11, 1.2, 2)), 0)
def test_no_memory_used(self):
"""Make sure float_range response isn't a giant list of numbers."""
response = float_range2(0, 1024, 2**-4)
if isinstance(response, Generator):
next(response)
size = sum(
sys.getsizeof(obj)
for obj in response.gi_frame.f_locals.values())
else:
size = sys.getsizeof(response)
self.assertLess(size, 8000, 'Too much memory used')
self.assertNotEqual(type(response), list)
self.assertNotEqual(type(response), tuple)
def test_has_iterability(self):
self.assertEqual(list(float_range2(1, 11, 2)), [1, 3, 5, 7, 9])
self.assertEqual(list(float_range2(0.5, 7, 0.75)),
[0.5, 1.25, 2.0, 2.75, 3.5, 4.25, 5.0, 5.75, 6.5])
def test_equality(self):
self.assertEqual(float_range2(0, 5, 0.5), float_range2(0, 5, 0.5))
self.assertEqual(float_range2(5, 5), float_range2(10, 10))
self.assertEqual(float_range2(5, 11, 5), float_range2(5, 12, 5))
self.assertEqual(float_range2(10), float_range2(0, 10))
self.assertNotEqual(
float_range2(0, 2**10, 2**-10),
float_range2(0, 2**10 + 1, 2**-10),
)
self.assertEqual(float_range2(1000000), range(1000000))
self.assertEqual(range(1000000), float_range2(1000000))
self.assertFalse(float_range2(0, 5, 0.5) != float_range2(0, 5, 0.5))
class EqualToEverything:
def __eq__(self, other):
return True
self.assertEqual(float_range2(1000000), EqualToEverything())
def test_reversed(self):
r = reversed(float_range2(0.5, 7, 0.75))
self.assertEqual(list(r),
[6.5, 5.75, 5.0, 4.25, 3.5, 2.75, 2.0, 1.25, 0.5])
big_num = 1000000
self.assertEqual(next(reversed(float_range2(big_num))), big_num - 1)
def test_too_many_arguments(self):
with self.assertRaises(TypeError):
float_range2(0, 5, 1, 1)
with self.assertRaises(TypeError):
float_range2(0, 5, 1, 1, 1)
def test_no_arguments(self):
with self.assertRaises(TypeError):
float_range2()
def test_fractional_step_size(self):
self.assertEqual(list(float_range2(1, 6, 0.5)),
[1, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 4.5, 5.0, 5.5])
self.assertEqual(list(float_range2(1, 5.6, 0.5)),
[1, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 4.5, 5.0, 5.5])
def test_optional_start(self):
self.assertEqual(list(float_range2(0, 6)), [0, 1, 2, 3, 4, 5])
self.assertEqual(list(float_range2(6)), [0, 1, 2, 3, 4, 5])
self.assertEqual(list(float_range2(4.2)), [0, 1, 2, 3, 4])
def test_optional_step(self):
self.assertEqual(list(float_range2(1, 6, 1)), [1, 2, 3, 4, 5])
self.assertEqual(list(float_range2(1, 6)), [1, 2, 3, 4, 5])
self.assertEqual(list(float_range2(0.5, 6)),
[0.5, 1.5, 2.5, 3.5, 4.5, 5.5])
