diff --git a/common/lib/xmodule/xmodule/tests/test_logic.py b/common/lib/xmodule/xmodule/tests/test_logic.py
index 6cd46a26ee..e60af63921 100644
--- a/common/lib/xmodule/xmodule/tests/test_logic.py
+++ b/common/lib/xmodule/xmodule/tests/test_logic.py
@@ -4,10 +4,12 @@
import json
import unittest
+from lxml import etree
+
from xmodule.poll_module import PollDescriptor
from xmodule.conditional_module import ConditionalDescriptor
from xmodule.word_cloud_module import WordCloudDescriptor
-
+from xmodule.videoalpha_module import VideoAlphaDescriptor
class PostData:
"""Class which emulate postdata."""
@@ -117,3 +119,33 @@ class WordCloudModuleTest(LogicTest):
)
self.assertEqual(100.0, sum(i['percent'] for i in response['top_words']) )
+
+
+class VideoAlphaModuleTest(LogicTest):
+ descriptor_class = VideoAlphaDescriptor
+
+ raw_model_data = {
+ 'data': ''
+ }
+
+ def test_get_timeframe_no_parameters(self):
+ xmltree = etree.fromstring('test')
+ output = self.xmodule._get_timeframe(xmltree)
+ self.assertEqual(output, ('', ''))
+
+ def test_get_timeframe_with_one_parameter(self):
+ xmltree = etree.fromstring(
+ 'test'
+ )
+ output = self.xmodule._get_timeframe(xmltree)
+ self.assertEqual(output, (247, ''))
+
+ def test_get_timeframe_with_two_parameters(self):
+ xmltree = etree.fromstring(
+ '''test'''
+ )
+ output = self.xmodule._get_timeframe(xmltree)
+ self.assertEqual(output, (247, 47079))
diff --git a/common/lib/xmodule/xmodule/videoalpha_module.py b/common/lib/xmodule/xmodule/videoalpha_module.py
index 83d0b4badd..16230480a7 100644
--- a/common/lib/xmodule/xmodule/videoalpha_module.py
+++ b/common/lib/xmodule/xmodule/videoalpha_module.py
@@ -103,9 +103,11 @@ class VideoAlphaModule(VideoAlphaFields, XModule):
return ''
else:
x = time.strptime(s, '%H:%M:%S')
- return datetime.timedelta(hours=x.tm_hour,
- minutes=x.tm_min,
- seconds=x.tm_sec).total_seconds()
+ return datetime.timedelta(
+ hours=x.tm_hour,
+ minutes=x.tm_min,
+ seconds=x.tm_sec
+ ).total_seconds()
return parse_time(xmltree.get('start_time')), parse_time(xmltree.get('end_time'))