samila/test/error_test.py

27 wiersze
1.0 KiB
Python

# -*- coding: utf-8 -*-
"""
>>> from samila import *
>>> import os
>>> from pytest import warns
>>> g = GenerativeImage(data="data.json")
Traceback (most recent call last):
...
samila.errors.samilaDataError: Provided data file is not supported. It should be either file or io.IOBase.
>>> with open('data.json', 'w') as fp:
... result = fp.write('test')
>>> g = GenerativeImage(data=open("data.json", 'r'))
Traceback (most recent call last):
...
samila.errors.samilaDataError: Provided data format is wrong. It should be in JSON format including data1 and data2 fields.
>>> g = GenerativeImage(lambda x,y: 0, lambda x,y: 0)
>>> g.generate(step=0.1)
>>> result = g.save_data('data.json')
>>> with warns(RuntimeWarning, match="Just data is provided, generate method is not available in this mode."):
... g = GenerativeImage(data=open('data.json', 'r'))
>>> g.generate()
Traceback (most recent call last):
...
samila.errors.samilaGenerateError: At least one of the given functions are None.
>>> os.remove('data.json')
"""