def test_optional_step(self):
self.assertEqual(list(float_range(1, 6, 1)), [1, 2, 3, 4, 5])
self.assertEqual(list(float_range(1, 6)), [1, 2, 3, 4, 5])
self.assertEqual(
list(float_range(0.5, 6)),
[0.5, 1.5, 2.5, 3.5, 4.5, 5.5]
)
def test_optional_start(self):
self.assertEqual(list(float_range(0, 6)), [0, 1, 2, 3, 4, 5])
self.assertEqual(list(float_range(6)), [0, 1, 2, 3, 4, 5])
self.assertEqual(
list(float_range(4.2)),
[0, 1, 2, 3, 4]
)
def test_reversed(self):
r = reversed(float_range(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_range(big_num))), big_num - 1)
self.assertEqual(list(reversed(float_range(4, 0, -1))), [1, 2, 3, 4])
def test_fractional_step_size(self):
self.assertEqual(
list(float_range(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_range(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_negative_step(self):
with self.assertRaises(StopIteration):
# Should be empty so StopIteration should be raised
next(iter(float_range(1, 6, -1)))
self.assertEqual(list(float_range(5, 0, -1)), [5, 4, 3, 2, 1])
self.assertEqual(list(float_range(0.5, 6)),
[0.5, 1.5, 2.5, 3.5, 4.5, 5.5])
self.assertEqual(list(float_range(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_can_be_indexed(self):
r = float_range(0.5, 7, 0.75)
self.assertEqual(r[0], 0.5)
self.assertEqual(r[1], 1.25)
self.assertEqual(r[3], 2.75)
self.assertEqual(r[6], 5.0)
self.assertEqual(r[8], 6.5)
self.assertEqual(r[-1], 6.5)
self.assertEqual(r[-3], 5.0)
self.assertEqual(r[-6], 2.75)
self.assertEqual(r[-9], 0.5)
with self.assertRaises(IndexError):
r[9]
with self.assertRaises(IndexError):
r[-10]
self.assertEqual(float_range(5, 0, -1)[1], 4)
def test_can_be_sliced(self):
r = float_range(0.5, 7, 0.75)
self.assertEqual(list(r[0:2]), [0.5, 1.25])
self.assertEqual(list(r[:2]), [0.5, 1.25])
self.assertEqual(list(r[-3:]), [5.0, 5.75, 6.5])
self.assertEqual(list(r[-100:0]), [])
self.assertEqual(list(r[-1:100]), [6.5])
self.assertEqual(list(r[::2]), [0.5, 2.0, 3.5, 5.0, 6.5])
self.assertEqual(list(r[len(r)::]), [])
def test_no_memory_used(self):
"""Make sure float_range response isn't a giant list of numbers."""
response = float_range(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_can_be_looped_over_multiple_times(self):
expected = [0.5, 1.25, 2.0, 2.75, 3.5, 4.25, 5.0, 5.75, 6.5]
output = float_range(0.5, 7, 0.75)
self.assertEqual(list(output), list(output))
self.assertEqual(list(output), expected)
def test_has_length(self):
self.assertEqual(len(float_range(100)), 100)
self.assertEqual(len(float_range(1, 100)), 99)
self.assertEqual(len(float_range(1, 11, 2)), 5)
self.assertEqual(len(float_range(0.5, 7, 0.75)), 9)
self.assertEqual(len(float_range(1000000)), 1000000)
def test_too_many_arguments(self):
with self.assertRaises(TypeError):
float_range(0, 5, 1, 1)
with self.assertRaises(TypeError):
float_range(0, 5, 1, 1, 1)
def test_no_arguments(self):
with self.assertRaises(TypeError):
float_range()
def test_has_iterability(self):
self.assertEqual(list(float_range(1, 11, 2)), [1, 3, 5, 7, 9])
self.assertEqual(
list(float_range(0.5, 7, 0.75)),
[0.5, 1.25, 2.0, 2.75, 3.5, 4.25, 5.0, 5.75, 6.5]
)
