from knr_ownership_migration import calculate_time_taken

class TestKnrOwnershipMigration:
    def test_calculate_time_taken(self):
        assert calculate_time_taken(2, 32) == '30s', 'Formats time taken in seconds'
        assert calculate_time_taken(5, 605) == '10m 0s', 'Formats time taken in minutes'
        assert calculate_time_taken(2, 307) == '5m 5s', 'Formats time taken in minutes and seconds'