"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "57f9f9dcf94777ad0ac26c78287c10edcb1ff264"
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): ...@@ -36,6 +36,13 @@ class Event(object):
self._duration = duration self._duration = duration
self._t0 = t0 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 @property
def period(self): def period(self):
return self._period return self._period
......
...@@ -25,6 +25,13 @@ from light_curve_util.periodic_event import Event ...@@ -25,6 +25,13 @@ from light_curve_util.periodic_event import Event
class EventTest(absltest.TestCase): 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): def testEquals(self):
event = Event(period=100, duration=5, t0=2) event = Event(period=100, duration=5, t0=2)
...@@ -72,5 +79,5 @@ class EventTest(absltest.TestCase): ...@@ -72,5 +79,5 @@ class EventTest(absltest.TestCase):
event.equals(Event(period=100, duration=5, t0=10), t0_durations=2)) event.equals(Event(period=100, duration=5, t0=10), t0_durations=2))
if __name__ == '__main__': if __name__ == "__main__":
absltest.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