[파이썬] 문제 : 계산기 클래스 만들기

import math                                        # math 패키지 추가

class Calc:
    def __init__(self, val1, val2):
        self.val1 = val1
        self.val2 = val2

    def setData(self, val1, val2):
        self.val1 = val1
        self.val2 = val2

    def add(self):
        return self.val1 + self.val2
    
    def sub(self):
        return self.val1 - self.val2
    
    def mul(self):
        return self.val1 * self.val2
    
    def div(self):
        return self.val1 / self.val2

# Val1 = Calc(5, 8)
# print(Val1.add(), Val1.sub(), Val1.mul(), Val1.div())
      
class FinalCalc(Calc):                                 # Calc 클래스 상속 받음.
    def Combi(self):  # n : val1 , r : val2             # 이항계수 함수 추가
        if self.val2 <= 0 or self.val2 > self.val1:     # r > n 또는 0 >= r 인 경우 0
            return 0
        
        return math.comb(self.val1, self.val2)          # math.comb() 함수 사용
    
Val1 = FinalCalc(5, 3)
print(Val1.Combi())

Val1 = FinalCalc(8, 4)
print(Val1.Combi())