Hello, looking for help with a unittest issue I'm having.

Context

I'm building a calculator for the game Cities Skylines; Industry, which will calculate how many resources are being made and required for each industry area. The amounts grow based on multipliers for area level, number of barracks, budget and policy enabled. I'm writing this in pycharm and using the built in unittest library.

I have written a function for each multiplier and a function to calculate the total multiplier to be applied.

My tests for the individual multiplier functions are working, however my test for the total multiplier is failing, however when I print out the value of the total multiplier it matches my test assertion.

Test case

def test_total_multiplier(self):
    assert calculator.total_multiplier(TEST_AREA) == 1.0

Function trying to test

def total_multiplier(area):
    tm = round(level_multiplier(area) * budget_multiplier() * barracks_multiplier(area) * 
    automation_multiplier(area),
           2)
    if tm > 2.55:
        return 2.55
    else:
        return tm

Console message

1.0

....F

FAIL: test_total_multiplier (main.CalculatorTests)

Traceback (most recent call last): File "C:/Users/...", line 86, in test_total_multiplier assert calculator.total_multiplier(TEST_AREA) == 1.0 AssertionError

Ran 5 tests in 0.001s

FAILED (failures=1)

Process finished with exit code 1

First line of the console message is the result of this print;

print(calculator.total_multiplier(TEST_AREA))

As you can see the value matches, so I can't figure out why I'm getting the assertion error.

I suspect this has something to do with floats, but no clue how to resolve, any help would be appreciated.

Thanks, James.