def test_calculate_building(self):
"""should return a dictionary of nested arrays"""
calc = CodeCalculator(fans=6, floors=2)
code_dictionary = calc.calculate_building()
self.assertIsInstance(code_dictionary, dict)
def test_correct_number_of_codes_returned(self):
"""should return an array of 6 fan codes"""
calc = CodeCalculator(fans=6, floors=1)
code_array = calc.calculate_floor()
self.assertEqual(len(code_array), 6)
def test_calculate_floor_3_fans(self):
"""should return an array of arrays"""
calc = CodeCalculator(fans=6, floors=1)
code_array = calc.calculate_floor()
self.assertIsInstance(code_array[0], list)
self.assertIsInstance(code_array[1], list)
def test_consecutive_codes_are_different(self):
calc = CodeCalculator(fans=2, floors=1)
code_array = calc.calculate_floor()
code1 = code_array[0]
code2 = code_array[1]
self.assertNotEqual(code1, code2)
def test_there_are_no_vertical_collisions(self):
calc = CodeCalculator(fans=6, floors=3)
code_dictionary = calc.calculate_building()
# not wild about hardcoding my indices, ideally
# I would create a bldg and fl object
# that is iterable
higher_floor_index = 1
lower_floor_index = 2
while lower_floor_index < len(code_dictionary):
fan_index = 0
for fan_code in code_dictionary[f"fl{higher_floor_index}"]:
print(f"this is the fan index: {fan_index}")
self.assertNotEqual(
fan_code,
code_dictionary[f"fl{lower_floor_index}"][fan_index])
fan_index += 1
higher_floor_index += 1
lower_floor_index += 1
def test_adjacent_fans_use_different_codes(self):
calc = CodeCalculator(fans=2, floors=1)
codes = calc.calculate_floor()
code1 = codes[0]
code2 = codes[1]
self.assertNotEqual(code1, code2)
def test_codes_have_correct_format(self):
calc = CodeCalculator(fans=1, floors=1)
code_array = calc.calculate_floor()
self.assertIsInstance(code_array, list)