def test_recalculate_heat(): simple_map = zeros8((3, 4)) b = BeeClust(simple_map) b.heatmap[:, :] = 100 assert numpy.isclose(b.heatmap, 100).all() b.recalculate_heat() assert numpy.isclose(b.heatmap, T_ENV).all() b.map[:, :] = HEATER b.recalculate_heat() assert numpy.isclose(b.heatmap, T_HEATER).all()
def test_recalculate_heat_is_callable(): b = BeeClust(zeros8((2, 2))) b.recalculate_heat()