Allow number for path (#4)

* Allow number for path

* Fix typos

* Add more tests

README.md

@@ -16,7 +16,7 @@ set(object, path, value) --> object
 delete(object, path) --> object
 ```
 
-None of the function mutate  the input object. For efficiency the returned object is not a deep clone of the original, but a shallow copy of the objects in the mutated path.
+None of the functions mutate the input object. For efficiency the returned object is not a deep clone of the original, but a shallow copy of the objects in the mutated path.
 
 
 ## Usage

index.js

@@ -1,7 +1,7 @@
 'use strict';
 
 module.exports.set = function(obj, prop, value) {
-	prop = typeof prop === 'string' ? propToArray(prop) : prop;
+	prop = typeof prop === 'number' ? propToArray(prop.toString()) : typeof prop === 'string' ? propToArray(prop) : prop;
 
 	var setPropImmutableRec = function(obj, prop, value, i) {
 		var clone, head = prop[i];
@@ -24,7 +24,7 @@ module.exports.set = function(obj, prop, value) {
 };
 
 module.exports.get = function(obj, prop) {
-	prop = typeof prop === 'string' ? propToArray(prop) : prop;
+	prop = typeof prop === 'number' ? propToArray(prop.toString()) : typeof prop === 'string' ? propToArray(prop) : prop;
 
 	for (var i = 0; i < prop.length; i++) {
 		if (typeof obj !== 'object') {
@@ -41,7 +41,7 @@ module.exports.get = function(obj, prop) {
 };
 
 module.exports.delete = function(obj, prop) {
-	prop = typeof prop === 'string' ? propToArray(prop) : prop;
+	prop = typeof prop === 'number' ? propToArray(prop.toString()) : typeof prop === 'string' ? propToArray(prop) : prop;
 
 	var deletePropImmutableRec = function(obj, prop, i) {
 		var clone, head = prop[i];

test/dot-prop-immutable-number.spec.js

@@ -0,0 +1,62 @@
+var dotProp = require('..');
+
+describe('dot-prop-immutable.number.spec.js', function () {
+
+	var arr = [1, {a: false}];
+
+	var result;
+	describe('when have an array', () => {
+
+		describe('when set prop using number as path', () => {
+
+			before(function () {
+				result = dotProp.set(arr, 1, 3);
+			});
+
+			it('should replace prop', () => {
+				expect(result).to.eql([
+				1,
+				3
+				]);
+			});
+
+			it('invariant', arrInvariant);
+		});
+
+
+		describe('when get prop using number as path', () => {
+
+			before(function () {
+				result = dotProp.get(arr, 1);
+			});
+
+			it('should get prop', () => {
+				expect(result).to.eql({a: false});
+			});
+
+			it('invariant', arrInvariant);
+		});
+
+		describe('when delete prop using number as path', () => {
+
+			before(function () {
+				result = dotProp.delete(arr, 1);
+			});
+
+			it('should remove prop', () => {
+				expect(result).to.eql([
+				1
+				]);
+			});
+			it('invariant', arrInvariant);
+		});
+	});
+
+	function arrInvariant() {
+		expect(arr).to.eql(
+		[1, {a: false}]
+		);
+	}
+});
+
+