input: Write a function to implement an OR gate using the given code.
  assert or_gate(0, 0) == 0
  assert or_gate(0, 1) == 1
  assert or_gate(1, 0) == 1
  assert or_gate(1, 1) == 1
output: def or_gate(a, b):
      return a | b