from linegenerator import LineGenerator
import math as m


class StepLineGenerator(LineGenerator):

    def __init__(self, name, units, start, end, step):

        num = int(m.floor((end - start)/step))

        super(StepLineGenerator, self).__init__(name, units, start, step, num)

from linegenerator import LineGenerator


class StepLineGenerator(LineGenerator):

    def __init__(self, name, units, start, end, step):

        num = int((end - start)/step) + 1

        super(StepLineGenerator, self).__init__(name, units, start, step, num)

Add extra point to include start
Add extra point to include start

  from linegenerator import LineGenerator
- import math as m
  
  
  class StepLineGenerator(LineGenerator):
  
      def __init__(self, name, units, start, end, step):
  
-         num = int(m.floor((end - start)/step))
+         num = int((end - start)/step) + 1
  
          super(StepLineGenerator, self).__init__(name, units, start, step, num)
  
Add extra point to include start
## Code Before:
from linegenerator import LineGenerator
import math as m


class StepLineGenerator(LineGenerator):

    def __init__(self, name, units, start, end, step):

        num = int(m.floor((end - start)/step))

        super(StepLineGenerator, self).__init__(name, units, start, step, num)

## Instruction:
Add extra point to include start
## Code After:
from linegenerator import LineGenerator


class StepLineGenerator(LineGenerator):

    def __init__(self, name, units, start, end, step):

        num = int((end - start)/step) + 1

        super(StepLineGenerator, self).__init__(name, units, start, step, num)
