Fix Test cases

test/elements/Image.test.tsx

@@ -5,8 +5,11 @@ import { Image } from './../../src/elements/Image';
 
 describe('Image', () => {
     it('should render a figure with .image', () => {
-        const component = shallow(<Image>My Image</Image>);
-        expect(component.contains(<figure className='image'>My Image</figure>)).toBe(true);
+        const component = shallow(<Image src='something' />);
+        expect(component.is('figure')).toBe(true);
+        expect(component.hasClass('image')).toBe(true);
+        expect(component.children().is('img')).toBe(true);
+        expect(component.children().contains(<img src='something' />)).toBe(true);
     });
 
     it('should render a figure with .image and Square modifiers', () => {

test/grid/Tile.test.tsx

@@ -4,6 +4,19 @@ import { shallow } from 'enzyme';
 import { Tile } from './../../src/grid/Tile';
 
 describe('Tile', () => {
+    it('should render a custom component with Tile props', () => {
+        const customComponent = (props) => (
+            <div {...props}> My Tile </div>
+        );
+
+        const customComponentRendered = (
+            <div className='tile'> My Tile </div>
+        );
+
+        const component = shallow(<Tile render={customComponent} />);
+        expect(component.contains(customComponentRendered)).toBe(true);
+    });
+
     it('should render a div with .tile', () => {
         const component = shallow(<Tile>Any Content</Tile>);
         expect(component.contains(<div className='tile'>Any Content</div>)).toBe(true);