def test_if_data_type_of_vehicle_brands_generator_return_is_string(self):
brands = vehicle_brand(n=87)
for brand in brands:
self.assertIsInstance(brand, str)
def test_if_vehicle_brand_generator_not_exceed_min_and_max_limit_of_return(
self):
with self.assertRaises(ValueError):
vehicle_brand(n=0)
vehicle_brand(n=-1)
vehicle_brand(n=88)
def test_vehicle_brand_generator_with_data_only_argument_as_false(self):
brands = vehicle_brand(data_only=False)
self.assertIsInstance(brands, dict)
self.assertCountEqual(['msg', 'data'], brands.keys())
self.assertEqual(brands['msg'], 'success')
self.assertIsInstance(brands['data'], list)
def test_if_vehicle_brand_generator_returns_max_and_min_number_of_data(
self):
min_brands = vehicle_brand(n=1)
max_brands = vehicle_brand(n=87)
self.assertGreaterEqual(len(min_brands), 1)
self.assertLessEqual(len(max_brands), 87)
def test_vehicle_brand_generator_with_data_only_argument_as_true(self):
brands = vehicle_brand(data_only=True)
self.assertIsInstance(brands, list)