⚡ Add support for copying of DayuTimeCode objects

goodbyekansas · Jun 15, 2020 · cea6c3e · cea6c3e
1 parent 1c58a17
commit cea6c3e
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -4,4 +4,5 @@ python:
 install:
         - pip install -r requirements.txt
 script:
+        - export PYTHONPATH=$PYTHONPATH:$(pwd)
         - pytest
diff --git a/dayu_timecode/base.py b/dayu_timecode/base.py
@@ -247,6 +247,11 @@ def __int__(self):
     def __neg__(self):
         self.time = -self.time
         return self
+
+    def __copy__(self):
+        dayu_tc = type(self)(self.frame(), fps=self.fps)
+        dayu_tc.__dict__.update(self.__dict__)
+        return dayu_tc
 
     def retime(self, point, retime):
         '''

diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_dayuTimeCode.py b/tests/test_dayuTimeCode.py
@@ -4,6 +4,7 @@
 __author__ = 'andyguo'
 
 from unittest import TestCase
+import copy
 from dayu_timecode.base import DayuTimeCode, Fraction
 from dayu_timecode.error import *
 from dayu_timecode.config import *
@@ -308,3 +309,13 @@ def test_retime(self):
         self.assertEqual(DayuTimeCode(1440).retime(0, 2.0), 720)
         self.assertEqual(DayuTimeCode(1440).retime(-100, 0.5), 2980)
         self.assertEqual(DayuTimeCode(1440).retime(2880, 2), 2160)
+
+    def test___copy__(self):
+        a = DayuTimeCode(1440, fps=25)
+        b = a
+        a.fps = 50
+        self.assertEqual(b.fps, 50)    
+        c = copy.copy(a)
+        c.fps = 25
+        self.assertNotEqual(c.fps, a.fps)
+        self.assertNotEqual(c.fps, b.fps)