Карта сокровищ - Python - Ответ 13230333

import unittest
 
class TestTreasure(unittest.TestCase):
 
    def test_one_treasure(self):
        # Arrange
        treasure = Treasure()
        input = [(15, 16)]
        expectedAmount = 1
 
        # Act
        actualAmount = treasure.GetAmount(input)
 
        # Assert
        self.assertEqual(expectedAmount, actualAmount)

class Treasure:
    def GetAmount(self, input):
        return -1

import unittest
 
from treasure import Treasure

    self.assertEqual(expectedAmount, actualAmount)
AssertionError: 1 != -1

class Treasure:
    def GetAmount(self, input):
        if len(input) == 1:
            return 1
        return -1

import unittest
 
from treasure import Treasure
 
class TestTreasure(unittest.TestCase):
 
    def test_one_treasure(self):
        # Arrange
        treasure = Treasure()
        input = [(15, 16)]
        expectedAmount = 1
 
        # Act
        actualAmount = treasure.GetAmount(input)
 
        # Assert
        self.assertEqual(expectedAmount, actualAmount)
 
    def test_two_treasures(self):
        # Arrange
        treasure = Treasure()
        input = [(15, 16), (16, 18)]
        expectedAmount = 2
 
        # Act
        actualAmount = treasure.GetAmount(input)
 
        # Assert
        self.assertEqual(expectedAmount, actualAmount)

test_treasure.py", line 29, in test_two_treasures
    self.assertEqual(expectedAmount, actualAmount)
AssertionError: 2 != -1