Initial commit

Author: jia3ep

.gitignore

@@ -0,0 +1,34 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+tests/__pycache__/
+*.py[cod]
+*$py.class
+*.swp
+
+# IDE files
+.idea/
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+.DS_Store
+.coverage
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+

LICENSE

@@ -0,0 +1,28 @@
+Copyright (c) 2016, Kirill V. Lyadvinsky
+http://www.codeatcpp.com
+
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification,
+are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+3. Neither the name of the copyright holder nor the names of its contributors
+   may be used to endorse or promote products derived from this software without
+   specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+

Makefile

@@ -0,0 +1,20 @@
+.PHONY: init test clean
+
+test:
+	python3 -m unittest discover -v
+
+init:
+	pip-3.2 install -r requirements.txt
+
+clean:
+	rm -rf test/*.pyc
+	rm -rf zxtools/*.pyc
+	rm -rf test/__pycache__
+	rm -rf zxtools/__pycache__
+
+coverage:
+	coverage run --source zxtools setup.py test
+	coverage report -m
+
+lint:
+	pylint zxtools -f parseable -r n

README.md

@@ -0,0 +1 @@
+numpy

requirements.txt

@@ -0,0 +1,25 @@
+#! /usr/bin/env python
+# vim: set fileencoding=utf-8 :
+
+from setuptools import setup, find_packages
+
+
+with open('README.md') as f:
+    readme = f.read()
+
+with open('LICENSE') as f:
+    license = f.read()
+
+setup(
+    name='zxtools',
+    version='0.0.1',
+    description='TR-DOS files utils',
+    long_description=readme,
+    author='Kirill V. Lyadvinsky',
+    author_email='[email protected]',
+    url='https://github.com/codeatcpp/zxtools',
+    license=license,
+    packages=find_packages(exclude=('test', 'docs')),
+    test_suite="test"
+)
+

setup.py

@@ -0,0 +1,114 @@
+#! /usr/bin/env python
+# vim: set fileencoding=utf-8 :
+""" hobeta.py tests """
+
+import os
+import io
+import struct
+import unittest
+import tempfile
+from collections import namedtuple
+from zxtools import hobeta
+
+
+class TestHobeta(unittest.TestCase):
+    def test_ckechsum(self):
+        data = b'F.load.AC\x00\x80\xf9\x06\x00\x07'
+        self.assertEqual(hobeta.calc_checksum(data), 20661)
+
+    def test_format_size(self):
+        header_len = struct.calcsize(hobeta.HEADER_FMT)
+        self.assertEqual(header_len, 17)
+
+    def test_format(self):
+        data = b"\x46\x2E\x6C\x6F\x61\x64\x2E\x41" \
+               b"\x43\x00\x80\xF9\x06\x00\x07\xB5\x50"
+        record = hobeta.Header._make(struct.unpack_from(hobeta.HEADER_FMT, data))
+
+        self.assertEqual(record.filename, b"F.load.A")
+        self.assertEqual(record.filetype, ord('C'))
+        self.assertEqual(record.start, 32768)
+        self.assertEqual(record.length, 1785)
+        self.assertEqual(record.first_sector, 0)
+        self.assertEqual(record.occupied_sectors, 7)
+        self.assertEqual(record.check_sum, 20661)
+
+    def test_parse_info(self):
+        test_file = io.BytesIO(b"\x46\x2E\x6C\x6F\x61\x64\x2E\x41\x43\x00\x80\xF9\x06\x00\x07\xB5"
+                               b"\x50\x00\x00\x3B\x20\x4C\x4F\x41\x44\x45\x52\x20\x66\x6F\x72\x20")
+        header, crc = hobeta.parse_info(test_file)
+
+        self.assertEqual(header.filename, b"F.load.A")
+        self.assertEqual(header.filetype, ord('C'))
+        self.assertEqual(header.start, 32768)
+        self.assertEqual(header.length, 1785)
+        self.assertEqual(header.first_sector, 0)
+        self.assertEqual(header.occupied_sectors, 7)
+        self.assertEqual(header.check_sum, 20661)
+        self.assertEqual(header.check_sum, crc)
+
+    def test_parse_info2(self):
+        test_file = io.BytesIO(b"\x46\x2E\x6C\x6F\x61\x64\x2E\x41\x43\x00\x80\x02\x00\x00\x07\xB5"
+                               b"\x50\x00\x00\x3B\x20\x4C\x4F\x41\x44\x45\x52\x20\x66\x6F\x72\x20")
+        header, crc = hobeta.parse_info(test_file)
+
+        self.assertEqual(header.filename, b"F.load.A")
+        self.assertEqual(header.filetype, ord('C'))
+        self.assertEqual(header.start, 32768)
+        self.assertEqual(header.length, 2)
+        self.assertEqual(header.first_sector, 0)
+        self.assertEqual(header.occupied_sectors, 7)
+        self.assertEqual(header.check_sum, 20661)
+        self.assertNotEqual(header.check_sum, crc)
+
+    def strip_header(self, test_input_file, ignore_header_info):
+        temp_output_path = tempfile.mkstemp()[1]
+        temp_output_file = open(temp_output_path, "wb")
+        Args = namedtuple('Args', "hobeta_file output_file ignore_header")
+        parsed_args = Args(test_input_file, temp_output_file, ignore_header_info)
+        copied_bytes = hobeta.strip_header(parsed_args)
+
+        return temp_output_path, copied_bytes
+
+    def test_strip_header1(self):
+        test_input_file = io.BytesIO(b"\x46\x2E\x6C\x6F\x61\x64\x2E\x41\x43\x00\x80\xF9\x06\x00\x07\xB5"
+                                     b"\x50\x00\x00\x3B\x20\x4C\x4F\x41\x44\x45\x52\x20\x66\x6F\x72\x20")
+        temp_output_path, bytes_count = self.strip_header(test_input_file, True)
+        try:
+            temp_output_file = open(temp_output_path, "rb")
+            temp_output_file.seek(0, os.SEEK_END)
+            self.assertEqual(temp_output_file.tell(), 15)
+            self.assertEqual(bytes_count, 15)
+        finally:
+            temp_output_file.close()
+            os.remove(temp_output_path)
+
+    def test_strip_header2(self):
+        test_input_file = io.BytesIO(b"\x46\x2E\x6C\x6F\x61\x64\x2E\x41\x43\x00\x80\xF9\x06\x00\x07\xB5"
+                                     b"\x50\x00\x00\x3B\x20\x4C\x4F\x41\x44\x45\x52\x20\x66\x6F\x72\x20")
+        temp_output_path, bytes_count = self.strip_header(test_input_file, False)
+        try:
+            temp_output_file = open(temp_output_path, "rb")
+            temp_output_file.seek(0, os.SEEK_END)
+            self.assertEqual(temp_output_file.tell(), 15)
+            self.assertEqual(bytes_count, 15)
+        finally:
+            temp_output_file.close()
+            os.remove(temp_output_path)
+
+    def test_strip_header3(self):
+        test_input_file = io.BytesIO(b"\x46\x2E\x6C\x6F\x61\x64\x2E\x41\x43\x00\x80\x0A\x00\x00\x07\xB5"
+                                     b"\x50\x00\x00\x3B\x20\x4C\x4F\x41\x44\x45\x52\x20\x66\x6F\x72\x20")
+        temp_output_path, bytes_count = self.strip_header(test_input_file, False)
+        try:
+            temp_output_file = open(temp_output_path, "rb")
+            temp_output_file.seek(0, os.SEEK_END)
+            self.assertEqual(temp_output_file.tell(), 10)
+            self.assertEqual(bytes_count, 10)
+        finally:
+            temp_output_file.close()
+            os.remove(temp_output_path)
+
+
+if __name__ == '__main__':
+    unittest.main()

test/__init__.py

@@ -0,0 +1,25 @@
+#! /usr/bin/env python
+# vim: set fileencoding=utf-8 :
+""" trdos.py tests """
+
+import unittest
+import struct
+from zxtools import trdos
+
+
+class TestTRDos(unittest.TestCase):
+    def test_fat_format(self):
+        data = b"filenameC\x00\x80\xf9\x06\x07\xC1\x01"
+        record = trdos.FATRecord._make(struct.unpack_from(trdos.FAT_RECORD_FMT, data))
+
+        self.assertEqual(record.filename, b"filename")
+        self.assertEqual(record.filetype, ord("C"))
+        self.assertEqual(record.start, 32768)
+        self.assertEqual(record.length, 1785)
+        self.assertEqual(record.occupied_sectors, 0x07)
+        self.assertEqual(record.first_sector, 0xC1)
+        self.assertEqual(record.first_track, 0x01)
+
+
+if __name__ == '__main__':
+    unittest.main()