Commit 3f5cef69 authored by Chris Shallue's avatar Chris Shallue Committed by Christopher Shallue
Browse files

Add __str__ and __repr__ to periodic_event.Event.

PiperOrigin-RevId: 202034513
parent f82ca018
......@@ -36,6 +36,13 @@ class Event(object):
self._duration = duration
self._t0 = t0
def __str__(self):
return "<period={}, duration={}, t0={}>".format(self.period, self.duration,
self.t0)
def __repr__(self):
return "Event({})".format(str(self))
@property
def period(self):
return self._period
......
......@@ -25,6 +25,13 @@ from light_curve_util.periodic_event import Event
class EventTest(absltest.TestCase):
def testStr(self):
self.assertEqual(str(Event(1, 2, 3)), "<period=1, duration=2, t0=3>")
def testRepr(self):
self.assertEqual(
repr(Event(1, 2, 3)), "Event(<period=1, duration=2, t0=3>)")
def testEquals(self):
event = Event(period=100, duration=5, t0=2)
......@@ -72,5 +79,5 @@ class EventTest(absltest.TestCase):
event.equals(Event(period=100, duration=5, t0=10), t0_durations=2))
if __name__ == '__main__':
if __name__ == "__main__":
absltest.main()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment