python_udacity/03_class_inheritance.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "03-class-inheritance.ipynb",
      "provenance": [],
      "authorship_tag": "ABX9TyN9v287FaETJsRW3H8P4/Ac",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/ferris77/my_notebooks/blob/main/python/03_class_inheritance.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "aeMTccpERdsT"
      },
      "source": [
        "This colab was made as part of one the OOP with Python during the Machine Learning Engineer Nanodegree. It creates one new class for the previous example.\n",
        "\n",
        "*  Student: Lucas Ferreira\n",
        "*  Date: Feb/2021"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "2yN838WZHMW9"
      },
      "source": [
        "class Clothing:\n",
        "\n",
        "    def __init__(self, color, size, style, price):\n",
        "        self.color = color\n",
        "        self.size = size\n",
        "        self.style = style\n",
        "        self.price = price\n",
        "        \n",
        "    def change_price(self, price):\n",
        "        self.price = price\n",
        "        \n",
        "    def calculate_discount(self, discount):\n",
        "        return self.price * (1 - discount)\n",
        "   \n",
        "    def calculate_shipping(self, weight, rate):\n",
        "        return weight * rate  \n",
        "\n",
        "class Shirt(Clothing):\n",
        "    \n",
        "    def __init__(self, color, size, style, price, long_or_short):\n",
        "        \n",
        "        Clothing.__init__(self, color, size, style, price)\n",
        "        self.long_or_short = long_or_short\n",
        "    \n",
        "    def double_price(self):\n",
        "        self.price = 2*self.price\n",
        "    \n",
        "class Pants(Clothing):\n",
        "\n",
        "    def __init__(self, color, size, style, price, waist):\n",
        "        \n",
        "        Clothing.__init__(self, color, size, style, price)\n",
        "        self.waist = waist\n",
        "        \n",
        "    def calculate_discount(self, discount):\n",
        "        return self.price * (1 - discount / 2)\n",
        "\n",
        "\n",
        "# TODO: Write a class called Blouse, that inherits from the Clothing class\n",
        "# and has the the following attributes and methods:\n",
        "#   attributes: color, size, style, price, country_of_origin\n",
        "#     where country_of_origin is a string that holds the name of a \n",
        "#     country\n",
        "#\n",
        "#   methods: triple_price, which has no inputs and returns three times\n",
        "#     the price of the blouse\n",
        "#\n",
        "#\n",
        "    \n",
        "class Blouse(Clothing):\n",
        "        \n",
        "    def __init__(self, color, size, style, price, country_of_origin):\n",
        "            \n",
        "        Clothing.__init__(self, color, size, style, price)\n",
        "        self.country_of_origin = country_of_origin\n",
        "            \n",
        "    def triple_price(self):\n",
        "        return self.price * 3\n",
        "    \n",
        "    \n",
        "# TODO: Add a method to the clothing class called calculate_shipping.\n",
        "#   The method has two inputs: weight and rate. Weight is a float\n",
        "#   representing the weight of the article of clothing. Rate is a float\n",
        "#   representing the shipping weight. The method returns weight * rate\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "yuRdFHyKL-WB",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "c3150293-0f17-41e9-e897-84e252618b42"
      },
      "source": [
        "# Unit tests to check your solution\n",
        "\n",
        "import unittest\n",
        "\n",
        "class TestClothingClass(unittest.TestCase):\n",
        "    def setUp(self):\n",
        "        self.clothing = Clothing('orange', 'M', 'stripes', 35)\n",
        "        self.blouse = Blouse('blue', 'M', 'luxury', 40, 'Brazil')\n",
        "        self.pants = Pants('black', 32, 'baggy', 60, 30)\n",
        "        \n",
        "    def test_initialization(self): \n",
        "        self.assertEqual(self.clothing.color, 'orange', 'color should be orange')\n",
        "        self.assertEqual(self.clothing.price, 35, 'incorrect price')\n",
        "        \n",
        "        self.assertEqual(self.blouse.color, 'blue', 'color should be blue')\n",
        "        self.assertEqual(self.blouse.size, 'M', 'incorrect size')\n",
        "        self.assertEqual(self.blouse.style, 'luxury', 'incorrect style')\n",
        "        self.assertEqual(self.blouse.price, 40, 'incorrect price')\n",
        "        self.assertEqual(self.blouse.country_of_origin, 'Brazil', 'incorrect country of origin')\n",
        "\n",
        "    def test_calculateshipping(self):\n",
        "        self.assertEqual(self.clothing.calculate_shipping(.5, 3), .5 * 3,\\\n",
        "         'Clothing shipping calculation not as expected') \n",
        "\n",
        "        self.assertEqual(self.blouse.calculate_shipping(.5, 3), .5 * 3,\\\n",
        "         'Clothing shipping calculation not as expected') \n",
        "    \n",
        "tests = TestClothingClass()\n",
        "\n",
        "tests_loaded = unittest.TestLoader().loadTestsFromModule(tests)\n",
        "\n",
        "unittest.TextTestRunner().run(tests_loaded)"
      ],
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "..\n",
            "----------------------------------------------------------------------\n",
            "Ran 2 tests in 0.002s\n",
            "\n",
            "OK\n"
          ],
          "name": "stderr"
        },
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "<unittest.runner.TextTestResult run=2 errors=0 failures=0>"
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 2
        }
      ]
    }
  ]
}