class NumbersTestCase(TestCase):
def setUp(self):
self.numbers = Numbers()
def tearDown(self):
del self.numbers
def test_floats(self):
result = self.numbers.floats()
self.assertEqual(len(result), 100)
self.assertIsInstance(result, array.array)
result = self.numbers.floats(n=3, to_list=True)
self.assertEqual(len(result), 1000)
self.assertIsInstance(result, list)
def test_primes(self):
result = self.numbers.primes()
self.assertEqual(len(result), 50)
self.assertIsInstance(result, array.array)
result = self.numbers.primes(n=3, to_list=True)
self.assertEqual(len(result), 500)
self.assertIsInstance(result, list)
def setUp(self):
self.numbers = Numbers()