<human>:  square wave with input and end.
<bot>: Here is a Python function to square wave with input and end

import numpy as np

def square_wave(input, end):
  """Square wave with input and end"""
  return (input - end) * (input - end) * (input - end) * (input - end)

# Enter the input
input = 1

# Enter the end
end = 0

# Calling the function
print((square_wave(input, end)))

#Example Output
# 1

#Remember to inspect code for bugs.