diff --git a/doc/example/report.md b/doc/example/report.md
index 567ea3a..aea0bbc 100644
--- a/doc/example/report.md
+++ b/doc/example/report.md
@@ -1,6 +1,8 @@
![memory map diagram](report.png)
|name|origin|size|remaining|collisions
|:-|:-|:-|:-|:-|
-|dtb|0x90|0x100|0x258|{}|
-|kernel|0x10|0x60|0x0|{}|
-|rootfs|0x70|0x10|0x10|{'kernel': '0x70'}|
+|kernel|0x10|0x50|-0x10|{'rootfs': '0x50'}|
+|uboot|0xD0|0x50|-0x10|{'uboot-scr': '0x110'}|
+|rootfs|0x50|0x30|0x10|{'kernel': '0x50'}|
+|dtb|0x90|0x30|0x10|{}|
+|uboot-scr|0x110|0x30|0x2a8|{'uboot': '0x110'}|
diff --git a/doc/example/report.png b/doc/example/report.png
index b3cb9be..19d6e23 100644
Binary files a/doc/example/report.png and b/doc/example/report.png differ
diff --git a/tests/test_docs.py b/tests/test_docs.py
index 8c790bf..0ac279c 100644
--- a/tests/test_docs.py
+++ b/tests/test_docs.py
@@ -4,20 +4,46 @@
def test_generate_doc_example():
- ''' should create custom report dir/files '''
+ """ """
with unittest.mock.patch('sys.argv',
['mmap_digram.diagram',
'kernel',
'0x10',
- '0x60',
+ '0x50',
'rootfs',
- '0x70',
- '0x10',
+ '0x50',
+ '0x30',
'dtb',
'0x90',
- '0x100',
+ '0x30',
+ 'uboot',
+ '0xD0',
+ '0x50',
+ 'uboot-scr',
+ '0x110',
+ '0x30',
"-o",
- "doc/example/report.md"]):
- mmdiagram.generator.Diagram()
- assert pathlib.Path("doc/example/report.md").exists()
- assert pathlib.Path("doc/example/report.png").exists()
+ "doc/example/report.md"],
+ mmdiagram.generator.height, 1000):
+ d = mmdiagram.generator.Diagram()
+ for region in d._region_list:
+ if region.name == "kernel":
+ assert region._origin == "0x10"
+ assert region._size == "0x50"
+ assert region.remain == "-0x10"
+ if region.name == "rootfs":
+ assert region._origin == "0x50"
+ assert region._size == "0x30"
+ assert region.remain == "0x10"
+ if region.name == "dtb":
+ assert region._origin == "0x90"
+ assert region._size == "0x30"
+ assert region.remain == "0x10"
+ if region.name == "uboot":
+ assert region._origin == "0xD0"
+ assert region._size == "0x50"
+ assert region.remain == "-0x10"
+ if region.name == "uboot-scr":
+ assert region._origin == "0x110"
+ assert region._size == "0x30"
+ assert region.remain == "0x2a8"